Wendong-Fan commited on
Commit
0b8f215
·
1 Parent(s): 4556f0e

remove hardcoded qwen model

Browse files
owl/camel/toolkits/video_analysis_toolkit.py CHANGED
@@ -28,12 +28,10 @@ from scenedetect.detectors import ( # type: ignore[import-untyped]
28
  )
29
 
30
  from camel.agents import ChatAgent
31
- from camel.configs import QwenConfig
32
  from camel.messages import BaseMessage
33
- from camel.models import ModelFactory, OpenAIAudioModels
34
  from camel.toolkits.base import BaseToolkit
35
  from camel.toolkits.function_tool import FunctionTool
36
- from camel.types import ModelPlatformType, ModelType
37
  from camel.utils import dependencies_required
38
 
39
  from .video_downloader_toolkit import (
@@ -101,6 +99,7 @@ class VideoAnalysisToolkit(BaseToolkit):
101
  def __init__(
102
  self,
103
  download_directory: Optional[str] = None,
 
104
  ) -> None:
105
  self._cleanup = download_directory is None
106
 
@@ -125,11 +124,7 @@ class VideoAnalysisToolkit(BaseToolkit):
125
 
126
  logger.info(f"Video will be downloaded to {self._download_directory}")
127
 
128
- self.vl_model = ModelFactory.create(
129
- model_platform=ModelPlatformType.QWEN,
130
- model_type=ModelType.QWEN_VL_MAX,
131
- model_config_dict=QwenConfig(temperature=0.2).as_dict(),
132
- )
133
 
134
  self.vl_agent = ChatAgent(
135
  model=self.vl_model, output_language="English"
 
28
  )
29
 
30
  from camel.agents import ChatAgent
 
31
  from camel.messages import BaseMessage
32
+ from camel.models import OpenAIAudioModels, BaseModelBackend
33
  from camel.toolkits.base import BaseToolkit
34
  from camel.toolkits.function_tool import FunctionTool
 
35
  from camel.utils import dependencies_required
36
 
37
  from .video_downloader_toolkit import (
 
99
  def __init__(
100
  self,
101
  download_directory: Optional[str] = None,
102
+ model: Optional[BaseModelBackend] = None,
103
  ) -> None:
104
  self._cleanup = download_directory is None
105
 
 
124
 
125
  logger.info(f"Video will be downloaded to {self._download_directory}")
126
 
127
+ self.vl_model = model
 
 
 
 
128
 
129
  self.vl_agent = ChatAgent(
130
  model=self.vl_model, output_language="English"