github/MemGPT
7
0
Fork 0
mirror of https://github.com/cpacker/MemGPT.git synced 2025-06-03 04:30:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

patch o4 streaming

Browse Source
This commit is contained in:
Sarah Wooders 2025-05-24 21:09:26 -07:00
parent f28fe24723
commit 61f8f7fedd
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
letta/interfaces/openai_streaming_interface.py
View File

@ -58,9 +58,10 @@ class OpenAIStreamingInterface:
def get_tool_call_object(self) -> ToolCall:
"""Useful for agent loop"""
function_name = self.last_flushed_function_name if self.last_flushed_function_name else self.function_name_buffer
return ToolCall(
id=self.letta_tool_message_id,
function=FunctionCall(arguments=self.current_function_arguments, name=self.last_flushed_function_name),
function=FunctionCall(arguments=self.current_function_arguments, name=function_name),
)
async def process(self, stream: AsyncStream[ChatCompletionChunk]) -> AsyncGenerator[LettaMessage, None]: