fix parser type hint

albertbou92 · albertbou92 · commit 3348a8dacf72 · 2025-11-20T14:20:33.000-08:00
diff --git a/packages/lmi/src/lmi/llms.py b/packages/lmi/src/lmi/llms.py
@@ -670,7 +670,7 @@ class LiteLLMModel(LLMModel):
             "Custom parser for converting LLM completions to tool requests. "
             "Accepts either `(completion: str, tools: list[dict] | None) -> ToolRequestMessage | Message` "
             "or `(choice: litellm.utils.Choices, tools: list[dict] | None) -> ToolRequestMessage | Message`. "
-            "Returns `ToolRequestMessage` on successful parsing, `Message` otherwise."
+            "Returns `aviary.core.ToolRequestMessage` on successful parsing, `aviary.core.Message` otherwise."
         ),
     )
     config: dict = Field(

Original file line number	Diff line number	Diff line change
`@@ -670,7 +670,7 @@ class LiteLLMModel(LLMModel):`
`670`	`670`	`"Custom parser for converting LLM completions to tool requests. "`
`671`	`671`	"Accepts either `(completion: str, tools: list[dict] \| None) -> ToolRequestMessage \| Message` "
`672`	`672`	"or `(choice: litellm.utils.Choices, tools: list[dict] \| None) -> ToolRequestMessage \| Message`. "
`673`		- "Returns `ToolRequestMessage` on successful parsing, `Message` otherwise."
	`673`	+ "Returns `aviary.core.ToolRequestMessage` on successful parsing, `aviary.core.Message` otherwise."
`674`	`674`	`),`
`675`	`675`	`)`
`676`	`676`	`config: dict = Field(`