Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat: added ability to disable the initial message sequence during agent creation (#1978)

Browse Source
This commit is contained in:
Charles Packer
2024-11-04 16:03:52 -08:00
committed by GitHub
parent 39999ce48c
commit b9f772f196
6 changed files with 136 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
letta/client/client.py
View File

@@ -376,6 +376,7 @@ class RESTClient(AbstractClient):
# metadata
metadata: Optional[Dict] = {"human:": DEFAULT_HUMAN, "persona": DEFAULT_PERSONA},
description: Optional[str] = None,
initial_message_sequence: Optional[List[Message]] = None,
) -> AgentState:
"""Create an agent
@@ -428,9 +429,18 @@ class RESTClient(AbstractClient):
agent_type=agent_type,
llm_config=llm_config if llm_config else self._default_llm_config,
embedding_config=embedding_config if embedding_config else self._default_embedding_config,
initial_message_sequence=initial_message_sequence,
)
# Use model_dump_json() instead of model_dump()
# If we use model_dump(), the datetime objects will not be serialized correctly
# response = requests.post(f"{self.base_url}/{self.api_prefix}/agents", json=request.model_dump(), headers=self.headers)
response = requests.post(
f"{self.base_url}/{self.api_prefix}/agents",
data=request.model_dump_json(), # Use model_dump_json() instead of json=model_dump()
headers={"Content-Type": "application/json", **self.headers},
)
response = requests.post(f"{self.base_url}/{self.api_prefix}/agents", json=request.model_dump(), headers=self.headers)
if response.status_code != 200:
raise ValueError(f"Status {response.status_code} - Failed to create agent: {response.text}")
return AgentState(**response.json())
@@ -1648,6 +1658,7 @@ class LocalClient(AbstractClient):
# metadata
metadata: Optional[Dict] = {"human:": DEFAULT_HUMAN, "persona": DEFAULT_PERSONA},
description: Optional[str] = None,
initial_message_sequence: Optional[List[Message]] = None,
) -> AgentState:
"""Create an agent
@@ -1702,6 +1713,7 @@ class LocalClient(AbstractClient):
agent_type=agent_type,
llm_config=llm_config if llm_config else self._default_llm_config,
embedding_config=embedding_config if embedding_config else self._default_embedding_config,
initial_message_sequence=initial_message_sequence,
),
actor=self.user,
)