Skip to content

Commit 4f0a4bf

Browse files
authored
chore: Update model_config (#3479)
1 parent af50bdf commit 4f0a4bf

17 files changed

+63
-29
lines changed

camel/configs/aihubmix_config.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,9 @@ class AihubMixConfig(BaseConfig):
8080
presence_penalty: Optional[float] = 0.0
8181
stream: Optional[bool] = False
8282
web_search_options: Optional[Dict] = None
83-
tool_choice: Optional[Union[Dict[str, str], str]] = None
83+
tool_choice: Optional[
84+
Union[Dict[str, Union[str, Dict[str, str]]], str]
85+
] = None
8486
parallel_tool_calls: Optional[bool] = None
8587
extra_headers: Optional[Dict[str, str]] = None
8688

camel/configs/bedrock_config.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,9 @@ class BedrockConfig(BaseConfig):
6666
top_p: Optional[float] = None
6767
top_k: Optional[int] = None
6868
stream: Optional[bool] = None
69-
tool_choice: Optional[Union[Dict[str, str], str]] = None
69+
tool_choice: Optional[
70+
Union[Dict[str, Union[str, Dict[str, str]]], str]
71+
] = None
7072
reasoning_effort: Optional[str] = None
7173

7274

camel/configs/cerebras_config.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,9 @@ class CerebrasConfig(BaseConfig):
8989
max_tokens: Optional[int] = None
9090
response_format: Optional[Dict] = None
9191
user: Optional[str] = None
92-
tool_choice: Optional[Union[Dict[str, str], str]] = None
92+
tool_choice: Optional[
93+
Union[Dict[str, Union[str, Dict[str, str]]], str]
94+
] = None
9395
reasoning_effort: Optional[str] = None
9496

9597

camel/configs/cometapi_config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
1414
from __future__ import annotations
1515

16-
from typing import Optional, Sequence, Union
16+
from typing import Dict, Optional, Sequence, Union
1717

1818
from camel.configs.base_config import BaseConfig
1919

@@ -98,7 +98,9 @@ class CometAPIConfig(BaseConfig):
9898
frequency_penalty: Optional[float] = None
9999
user: Optional[str] = None
100100
tools: Optional[list] = None
101-
tool_choice: Optional[Union[dict[str, str], str]] = None
101+
tool_choice: Optional[
102+
Union[Dict[str, Union[str, Dict[str, str]]], str]
103+
] = None
102104

103105

104106
COMETAPI_API_PARAMS = {param for param in CometAPIConfig.model_fields.keys()}

camel/configs/deepseek_config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515
from __future__ import annotations
1616

17-
from typing import Optional, Sequence, Type, Union
17+
from typing import Dict, Optional, Sequence, Type, Union
1818

1919
from pydantic import BaseModel
2020

@@ -93,7 +93,9 @@ class DeepSeekConfig(BaseConfig):
9393
presence_penalty: Optional[float] = None
9494
response_format: Optional[Union[Type[BaseModel], dict]] = None
9595
frequency_penalty: Optional[float] = None
96-
tool_choice: Optional[Union[dict[str, str], str]] = None
96+
tool_choice: Optional[
97+
Union[Dict[str, Union[str, Dict[str, str]]], str]
98+
] = None
9799
logprobs: Optional[bool] = None
98100
top_logprobs: Optional[int] = None
99101
stream_options: Optional[dict[str, bool]] = None

camel/configs/gemini_config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515
from __future__ import annotations
1616

17-
from typing import Optional, Sequence, Type, Union
17+
from typing import Dict, Optional, Sequence, Type, Union
1818

1919
from pydantic import BaseModel
2020

@@ -82,7 +82,9 @@ class GeminiConfig(BaseConfig):
8282
stop: Optional[Union[str, Sequence[str]]] = None
8383
max_tokens: Optional[int] = None
8484
response_format: Optional[Union[Type[BaseModel], dict]] = None
85-
tool_choice: Optional[Union[dict[str, str], str]] = None
85+
tool_choice: Optional[
86+
Union[Dict[str, Union[str, Dict[str, str]]], str]
87+
] = None
8688

8789

8890
Gemini_API_PARAMS = {param for param in GeminiConfig.model_fields.keys()}

camel/configs/groq_config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
1414
from __future__ import annotations
1515

16-
from typing import Optional, Sequence, Union
16+
from typing import Dict, Optional, Sequence, Union
1717

1818
from camel.configs.base_config import BaseConfig
1919

@@ -97,7 +97,9 @@ class GroqConfig(BaseConfig):
9797
response_format: Optional[dict] = None
9898
frequency_penalty: Optional[float] = None
9999
user: Optional[str] = None
100-
tool_choice: Optional[Union[dict[str, str], str]] = None
100+
tool_choice: Optional[
101+
Union[Dict[str, Union[str, Dict[str, str]]], str]
102+
] = None
101103

102104

103105
GROQ_API_PARAMS = {param for param in GroqConfig.model_fields.keys()}

camel/configs/internlm_config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# limitations under the License.
1313
# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
1414

15-
from typing import Optional, Union
15+
from typing import Dict, Optional, Union
1616

1717
from camel.configs.base_config import BaseConfig
1818

@@ -54,7 +54,9 @@ class InternLMConfig(BaseConfig):
5454
temperature: Optional[float] = None
5555
top_p: Optional[float] = None
5656
max_tokens: Optional[int] = None
57-
tool_choice: Optional[Union[dict[str, str], str]] = None
57+
tool_choice: Optional[
58+
Union[Dict[str, Union[str, Dict[str, str]]], str]
59+
] = None
5860

5961

6062
INTERNLM_API_PARAMS = {param for param in InternLMConfig.model_fields.keys()}

camel/configs/lmstudio_config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
1414
from __future__ import annotations
1515

16-
from typing import Optional, Sequence, Union
16+
from typing import Dict, Optional, Sequence, Union
1717

1818
from camel.configs.base_config import BaseConfig
1919

@@ -88,7 +88,9 @@ class LMStudioConfig(BaseConfig):
8888
presence_penalty: Optional[float] = None
8989
response_format: Optional[dict] = None
9090
frequency_penalty: Optional[float] = None
91-
tool_choice: Optional[Union[dict[str, str], str]] = None
91+
tool_choice: Optional[
92+
Union[Dict[str, Union[str, Dict[str, str]]], str]
93+
] = None
9294

9395

9496
LMSTUDIO_API_PARAMS = {param for param in LMStudioConfig.model_fields.keys()}

camel/configs/minimax_config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
1414
from __future__ import annotations
1515

16-
from typing import Optional, Sequence, Union
16+
from typing import Dict, Optional, Sequence, Union
1717

1818
from camel.configs.base_config import BaseConfig
1919

@@ -87,7 +87,9 @@ class MinimaxConfig(BaseConfig):
8787
max_tokens: Optional[int] = None
8888
response_format: Optional[dict] = None
8989
user: Optional[str] = None
90-
tool_choice: Optional[Union[dict[str, str], str]] = None
90+
tool_choice: Optional[
91+
Union[Dict[str, Union[str, Dict[str, str]]], str]
92+
] = None
9193

9294

9395
MINIMAX_API_PARAMS = {param for param in MinimaxConfig.model_fields.keys()}

0 commit comments

Comments
 (0)