opt

agent/video_preprocessor/video_preprocess.py

@@ -7,7 +7,8 @@ from typing import List, Optional, Union
 import json_repair
 from omagent_core.engine.worker.base import BaseWorker
 from omagent_core.models.asr.stt import STT
-from omagent_core.models.encoders.openai_encoder import OpenaiTextEmbeddingV3
+# from omagent_core.models.encoders.openai_encoder import OpenaiTextEmbeddingV3
+from omagent_core.models.encoders.azure_encoder import AzureTextEmbeddingV3
 from omagent_core.models.llms.base import BaseLLMBackend
 from omagent_core.models.llms.prompt import PromptTemplate
 from omagent_core.utils.registry import registry
@@ -33,7 +34,7 @@ class VideoPreprocessor(BaseLLMBackend, BaseWorker):
             ),
         ]
     )
-    text_encoder: OpenaiTextEmbeddingV3
+    text_encoder: AzureTextEmbeddingV3
 
     stt: STT
     scene_detect_threshold: Union[float, int] = 27

agent/video_qa/qa.py

@@ -8,7 +8,8 @@ from omagent_core.advanced_components.workflow.dnc.schemas.dnc_structure import
     TaskTree
 from omagent_core.engine.worker.base import BaseWorker
 from omagent_core.memories.ltms.ltm import LTM
-from omagent_core.models.encoders.openai_encoder import OpenaiTextEmbeddingV3
+# from omagent_core.models.encoders.openai_encoder import OpenaiTextEmbeddingV3
+from omagent_core.models.encoders.azure_encoder import AzureTextEmbeddingV3
 from omagent_core.models.llms.base import BaseLLMBackend
 from omagent_core.models.llms.prompt import PromptTemplate
 from omagent_core.utils.registry import registry
@@ -31,7 +32,7 @@ class VideoQA(BaseWorker, BaseLLMBackend):
             ),
         ]
     )
-    text_encoder: OpenaiTextEmbeddingV3
+    text_encoder: AzureTextEmbeddingV3
 
     def _run(self, video_md5: str, *args, **kwargs):
         self.stm(self.workflow_instance_id)["image_cache"] = {}