File tree Expand file tree Collapse file tree 5 files changed +18
-17
lines changed Expand file tree Collapse file tree 5 files changed +18
-17
lines changed Original file line number Diff line number Diff line change @@ -49,6 +49,7 @@ class ModelType(Enum):
49
49
CLAUDE_3_OPUS = "claude-3-opus-20240229"
50
50
CLAUDE_3_SONNET = "claude-3-sonnet-20240229"
51
51
CLAUDE_3_HAIKU = "claude-3-haiku-20240307"
52
+ CLAUDE_3_5_SONNET = "claude-3-5-sonnet-20240620"
52
53
53
54
# Nvidia models
54
55
NEMOTRON_4_REWARD = "nvidia/nemotron-4-340b-reward"
@@ -104,6 +105,7 @@ def is_anthropic(self) -> bool:
104
105
ModelType.CLAUDE_3_OPUS,
105
106
ModelType.CLAUDE_3_SONNET,
106
107
ModelType.CLAUDE_3_HAIKU,
108
+ ModelType.CLAUDE_3_5_SONNET,
107
109
}
108
110
109
111
@property
@@ -155,6 +157,7 @@ def token_limit(self) -> int:
155
157
ModelType.CLAUDE_3_OPUS,
156
158
ModelType.CLAUDE_3_SONNET,
157
159
ModelType.CLAUDE_3_HAIKU,
160
+ ModelType.CLAUDE_3_5_SONNET,
158
161
}:
159
162
return 200_000
160
163
elif self is ModelType.NEMOTRON_4_REWARD:
Original file line number Diff line number Diff line change 15
15
from camel.agents import ChatAgent
16
16
from camel.configs import ChatGPTConfig
17
17
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
+ )
19
26
20
27
# Define system message
21
28
sys_msg = BaseMessage.make_assistant_message(
22
29
role_name="Assistant",
23
30
content="You are a helpful assistant.",
24
31
)
25
32
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
-
31
33
# 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)
38
35
39
36
user_msg = BaseMessage.make_user_message(
40
37
role_name="User",
Original file line number Diff line number Diff line change @@ -35,7 +35,7 @@ colorama = "^0"
35
35
jsonschema = "^4"
36
36
protobuf = "^4"
37
37
pathlib = "^1.0.1"
38
- anthropic = "^0.28 .0"
38
+ anthropic = "^0.29 .0"
39
39
docstring-parser = "^0.15"
40
40
pydantic = ">=1.9,<3"
41
41
curl_cffi = "0.6.2"
Original file line number Diff line number Diff line change 31
31
ModelType.CLAUDE_3_OPUS,
32
32
ModelType.CLAUDE_3_SONNET,
33
33
ModelType.CLAUDE_3_HAIKU,
34
+ ModelType.CLAUDE_3_5_SONNET,
34
35
],
35
36
)
36
37
def test_anthropic_model(model_type):
You can’t perform that action at this time.
0 commit comments