Skip to content

Commit a42d029

Browse files
authored
feat: Add claude 3.5, update interface for zhipu example (#669)
1 parent 0e562bc commit a42d029

File tree

5 files changed

+18
-17
lines changed

5 files changed

+18
-17
lines changed

camel/types/enums.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ class ModelType(Enum):
4949
CLAUDE_3_OPUS = "claude-3-opus-20240229"
5050
CLAUDE_3_SONNET = "claude-3-sonnet-20240229"
5151
CLAUDE_3_HAIKU = "claude-3-haiku-20240307"
52+
CLAUDE_3_5_SONNET = "claude-3-5-sonnet-20240620"
5253

5354
# Nvidia models
5455
NEMOTRON_4_REWARD = "nvidia/nemotron-4-340b-reward"
@@ -104,6 +105,7 @@ def is_anthropic(self) -> bool:
104105
ModelType.CLAUDE_3_OPUS,
105106
ModelType.CLAUDE_3_SONNET,
106107
ModelType.CLAUDE_3_HAIKU,
108+
ModelType.CLAUDE_3_5_SONNET,
107109
}
108110

109111
@property
@@ -155,6 +157,7 @@ def token_limit(self) -> int:
155157
ModelType.CLAUDE_3_OPUS,
156158
ModelType.CLAUDE_3_SONNET,
157159
ModelType.CLAUDE_3_HAIKU,
160+
ModelType.CLAUDE_3_5_SONNET,
158161
}:
159162
return 200_000
160163
elif self is ModelType.NEMOTRON_4_REWARD:

examples/models/zhipuai_model_example copy.py renamed to examples/models/zhipuai_model_example.py

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -15,26 +15,23 @@
1515
from camel.agents import ChatAgent
1616
from camel.configs import ChatGPTConfig
1717
from camel.messages import BaseMessage
18-
from camel.types import ModelType
18+
from camel.models import ModelFactory
19+
from camel.types import ModelPlatformType, ModelType
20+
21+
model = ModelFactory.create(
22+
model_platform=ModelPlatformType.ZHIPU,
23+
model_type=ModelType.GLM_4,
24+
model_config=ChatGPTConfig(temperature=0.2),
25+
)
1926

2027
# Define system message
2128
sys_msg = BaseMessage.make_assistant_message(
2229
role_name="Assistant",
2330
content="You are a helpful assistant.",
2431
)
2532

26-
# Set model config
27-
model_config = ChatGPTConfig(
28-
temperature=0.2, top_p=0.9
29-
) # temperature=,top_p here can not be 1 or 0.
30-
3133
# Set agent
32-
camel_agent = ChatAgent(
33-
sys_msg,
34-
model_config=model_config,
35-
model_type=ModelType.GLM_4,
36-
)
37-
camel_agent.reset()
34+
camel_agent = ChatAgent(system_message=sys_msg, model=model)
3835

3936
user_msg = BaseMessage.make_user_message(
4037
role_name="User",

poetry.lock

Lines changed: 4 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ colorama = "^0"
3535
jsonschema = "^4"
3636
protobuf = "^4"
3737
pathlib = "^1.0.1"
38-
anthropic = "^0.28.0"
38+
anthropic = "^0.29.0"
3939
docstring-parser = "^0.15"
4040
pydantic = ">=1.9,<3"
4141
curl_cffi = "0.6.2"

test/models/test_anthropic_model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
ModelType.CLAUDE_3_OPUS,
3232
ModelType.CLAUDE_3_SONNET,
3333
ModelType.CLAUDE_3_HAIKU,
34+
ModelType.CLAUDE_3_5_SONNET,
3435
],
3536
)
3637
def test_anthropic_model(model_type):

0 commit comments

Comments
 (0)