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

fix: Fix create organization bug (#1956)

Browse Source
This commit is contained in:
Matthew Zhou
2024-10-30 13:55:48 -07:00
committed by GitHub
parent 7fa632aa94
commit 41e868c6cc
8 changed files with 99 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
letta/client/streaming.py
View File

@@ -13,6 +13,7 @@ from letta.schemas.letta_message import (
InternalMonologue,
)
from letta.schemas.letta_response import LettaStreamingResponse
from letta.schemas.usage import LettaUsageStatistics
def _sse_post(url: str, data: dict, headers: dict) -> Generator[LettaStreamingResponse, None, None]:
@@ -58,6 +59,8 @@ def _sse_post(url: str, data: dict, headers: dict) -> Generator[LettaStreamingRe
yield FunctionCallMessage(**chunk_data)
elif "function_return" in chunk_data:
yield FunctionReturn(**chunk_data)
elif "usage" in chunk_data:
yield LettaUsageStatistics(**chunk_data["usage"])
else:
raise ValueError(f"Unknown message type in chunk_data: {chunk_data}")