fixup type mismatches between rtvi data structures and together.py

This commit is contained in:
Kwindla Hultman Kramer
2024-08-27 17:39:21 -07:00
parent dfc2e62339
commit b9d362bd62
2 changed files with 4 additions and 3 deletions

View File

@@ -222,7 +222,7 @@ class RTVILLMFunctionCallResultData(BaseModel):
function_name: str
tool_call_id: str
arguments: dict
result: dict
result: dict | str
class RTVITranscriptionMessageData(BaseModel):

View File

@@ -173,7 +173,7 @@ class TogetherLLMService(LLMService):
try:
arguments = json.loads(args_string)
await self.call_function(context=context,
tool_call_id=uuid.uuid4(),
tool_call_id=str(uuid.uuid4()),
function_name=function_name,
arguments=arguments)
return
@@ -301,7 +301,8 @@ class TogetherAssistantContextAggregator(LLMAssistantContextAggregator):
self._function_call_result = None
self._context.add_message({
"role": "tool",
"content": frame.result
# Together expects the content here to be a string, so stringify it
"content": str(frame.result)
})
run_llm = True
else: