feat: require LLMConfig and EmbeddingConfig to be specified for agent creation + allow multiple simultaneous provider configs for server (#1814)

Co-authored-by: Shubham Naik <shubham.naik10@gmail.com>
Co-authored-by: Matthew Zhou <mattzh1314@gmail.com>
Co-authored-by: Matt Zhou <mattzhou@Matts-MacBook-Pro.local>
Co-authored-by: Shubham Naik <shub@memgpt.ai>
This commit is contained in:
Sarah Wooders
2024-10-04 19:35:00 -07:00
committed by GitHub
parent f9ec99b914
commit 9e4c7ad07f
31 changed files with 834 additions and 872 deletions

View File

@@ -11,6 +11,7 @@ from letta.schemas.enums import MessageRole
utils.DEBUG = True
from letta.config import LettaConfig
from letta.schemas.agent import CreateAgent
from letta.schemas.embedding_config import EmbeddingConfig
from letta.schemas.letta_message import (
AssistantMessage,
FunctionCallMessage,
@@ -20,6 +21,7 @@ from letta.schemas.letta_message import (
SystemMessage,
UserMessage,
)
from letta.schemas.llm_config import LLMConfig
from letta.schemas.memory import ChatMemory
from letta.schemas.message import Message
from letta.schemas.source import SourceCreate
@@ -73,7 +75,16 @@ def user_id(server):
def agent_id(server, user_id):
# create agent
agent_state = server.create_agent(
request=CreateAgent(name="test_agent", tools=BASE_TOOLS, memory=ChatMemory(human="Sarah", persona="I am a helpful assistant")),
request=CreateAgent(
name="test_agent",
tools=BASE_TOOLS,
memory=ChatMemory(
human="Sarah",
persona="I am a helpful assistant",
),
llm_config=LLMConfig.default_config("gpt-4"),
embedding_config=EmbeddingConfig.default_config(provider="openai"),
),
user_id=user_id,
)
print(f"Created agent\n{agent_state}")