refactor camel model

crab/agents/backend_models/camel_model.py

@@ -11,8 +11,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # =========== Copyright 2024 @ CAMEL-AI.org. All Rights Reserved. ===========
-import base64
-import io
 import json
 from typing import Any
 
@@ -34,28 +32,6 @@
     CAMEL_ENABLED = False
 
 
-def find_model_platform_type(model_platform_name: str) -> ModelPlatformType:
-    for platform in ModelPlatformType:
-        if platform.value.lower() == model_platform_name.lower():
-            return platform
-    all_models = [platform.value for platform in ModelPlatformType]
-    raise ValueError(
-        f"Model {model_platform_name} not found. Supported models are {all_models}"
-    )
-
-
-def find_model_type(model_name: str) -> ModelType | str:
-    for model in ModelType:
-        if model.value.lower() == model_name.lower():
-            return model
-    return model_name
-
-
-def decode_image(encoded_image: str) -> Image:
-    data = base64.b64decode(encoded_image)
-    return Image.open(io.BytesIO(data))
-
-
 class CamelModel(BackendModel):
     def __init__(
         self,
@@ -68,8 +44,8 @@ def __init__(
             raise ImportError("Please install camel-ai to use CamelModel")
         self.parameters = parameters or {}
         # TODO: a better way?
-        self.model_type = find_model_type(model)
-        self.model_platform_type = find_model_platform_type(model_platform)
+        self.model_type = self.find_model_type(model)
+        self.model_platform_type = self.find_model_platform_type(model_platform)
         self.client: ExternalToolAgent | None = None
         self.token_usage = 0
 
@@ -109,10 +85,27 @@ def reset(self, system_message: str, action_space: list[Action] | None) -> None:
         )
         self.token_usage = 0
 
+    @staticmethod
+    def find_model_platform_type(model_platform_name: str) -> "ModelPlatformType":
+        for platform in ModelPlatformType:
+            if platform.value.lower() == model_platform_name.lower():
+                return platform
+        all_models = [platform.value for platform in ModelPlatformType]
+        raise ValueError(
+            f"Model {model_platform_name} not found. Supported models are {all_models}"
+        )
+
+    @staticmethod
+    def find_model_type(model_name: str) -> "str | ModelType":
+        for model in ModelType:
+            if model.value.lower() == model_name.lower():
+                return model
+        return model_name
+
     @staticmethod
     def _convert_action_to_schema(
         action_space: list[Action] | None,
-    ) -> list[OpenAIFunction] | None:
+    ) -> "list[OpenAIFunction] | None":
         if action_space is None:
             return None
         return [OpenAIFunction(action.entry) for action in action_space]
@@ -147,12 +140,12 @@ def chat(self, messages: list[tuple[str, MessageType]]):
         )
         response = self.client.step(usermsg)
         self.token_usage += response.info["usage"]["total_tokens"]
-        tool_calls = response.info.get("tool_call_requests")
+        tool_call_request = response.info.get("tool_call_request")
 
         # TODO: delete this after record_message is refactored
         self.client.record_message(response.msg)
 
         return BackendOutput(
             message=response.msg.content,
-            action_list=self._convert_tool_calls_to_action_list(tool_calls),
+            action_list=self._convert_tool_calls_to_action_list([tool_call_request]),
         )