We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 84c2b38 commit d67b97cCopy full SHA for d67b97c
src/shelloracle/providers/ollama.py
@@ -78,6 +78,9 @@ async def generate(self, prompt: str) -> AsyncIterator[str]:
78
async with httpx.AsyncClient() as client:
79
async with client.stream("POST", self.endpoint, json=data, timeout=20.0) as stream:
80
async for line in stream.aiter_lines():
81
- yield json.loads(line)["response"]
+ response = json.loads(line)
82
+ if "error" in response:
83
+ raise ProviderError(response["error"])
84
+ yield response["response"]
85
except (httpx.HTTPError, httpx.StreamError) as e:
86
raise ProviderError(f"Something went wrong while querying Ollama: {e}") from e
0 commit comments