phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-02 00:34:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

disable assertion of dict type checking as an ad-hoc

Browse source
This commit is contained in:
3DAlgoLab 2024-12-01 10:16:54 +09:00
parent fe48b9fb8c
commit c486ece6fe
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/providers/remote/inference/ollama/ollama.py
View file

@ -269,7 +269,8 @@ class OllamaInferenceAdapter(Inference, ModelsProtocolPrivate):
if "messages" in params: if "messages" in params:
r = await self.client.chat(**params) r = await self.client.chat(**params)
else: else:
r = await self.client.generate(**params) r = await self.client.generate(**params)
r = dict(r) #ad-hoc
assert isinstance(r, dict) assert isinstance(r, dict)
if "message" in r: if "message" in r: