Spaces:

minchyeom
/

llmOS-Agent

Runtime error

minchyeom commited on Jun 1

Commit

79f2753

1 Parent(s): 832af4d

Add 'system' role to Msg TypedDict for enhanced message handling

Files changed (1) hide show

src/schema.py CHANGED Viewed

@@ -6,7 +6,7 @@ from ollama import Message
 class Msg(TypedDict, total=False):
-    role: Literal["user", "assistant", "tool"]
     content: str
     name: Optional[str]
     tool_calls: Optional[List[Message.ToolCall]]

 class Msg(TypedDict, total=False):
+    role: Literal["system", "user", "assistant", "tool"]
     content: str
     name: Optional[str]
     tool_calls: Optional[List[Message.ToolCall]]