opt

agent/video_preprocessor/webpage_vp.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 WebpageVideoPreprocessor(BaseLLMBackend, BaseWorker):
             ),
         ]
     )
-    text_encoder: OpenaiTextEmbeddingV3
+    text_encoder: AzureTextEmbeddingV3
     
     stt: STT
     scene_detect_threshold: Union[float, int] = 27

agent/video_qa/webpage_qa.py

@@ -5,7 +5,8 @@ import json_repair
 from pydantic import Field
 
 from omagent_core.engine.worker.base import BaseWorker
-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
@@ -26,7 +27,7 @@ class WebpageVideoQA(BaseWorker, BaseLLMBackend):
             ),
         ]
     )
-    text_encoder: OpenaiTextEmbeddingV3
+    text_encoder: AzureTextEmbeddingV3
     
     def _run(self, video_md5: str, video_path: str, instance_id: str, question: str, *args, **kwargs):
         self.stm(self.workflow_instance_id)["image_cache"] = {}