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
|
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 =
|
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"
|