From c486ece6fe4b792386795658f29dbec216e623f7 Mon Sep 17 00:00:00 2001 From: 3DAlgoLab Date: Sun, 1 Dec 2024 10:16:54 +0900 Subject: [PATCH] disable assertion of dict type checking as an ad-hoc --- llama_stack/providers/remote/inference/ollama/ollama.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/inference/ollama/ollama.py b/llama_stack/providers/remote/inference/ollama/ollama.py index 74c0b8601..68e03c0c9 100644 --- a/llama_stack/providers/remote/inference/ollama/ollama.py +++ b/llama_stack/providers/remote/inference/ollama/ollama.py @@ -269,7 +269,8 @@ class OllamaInferenceAdapter(Inference, ModelsProtocolPrivate): if "messages" in params: r = await self.client.chat(**params) else: - r = await self.client.generate(**params) + r = await self.client.generate(**params) + r = dict(r) #ad-hoc assert isinstance(r, dict) if "message" in r: