fix: resolve CI/CD issues for Python project
Some checks failed
CI / test (push) Has been cancelled
CI / lint (push) Has been cancelled
CI / type-check (push) Has been cancelled

This commit is contained in:
2026-02-04 11:16:17 +00:00
parent 87556d3699
commit ebcb09dd79

View File

@@ -1,6 +1,5 @@
"""Ollama client wrapper for ShellGenius."""
import json
import logging
from typing import Any, Dict, Generator, List, Optional
@@ -96,7 +95,7 @@ class OllamaClient:
"""
model = model or self.model
try:
response = self.client.generate(
response = self.client.generate( # type: ignore[call-overload]
model=model,
prompt=prompt,
stream=stream,
@@ -147,12 +146,13 @@ class OllamaClient:
"""
model = model or self.model
try:
response = self.client.chat(
response = self.client.chat( # type: ignore[call-overload]
model=model,
messages=messages,
stream=stream,
)
return {"success": True, "response": response}
return {"success": True, "response": response}
except Exception as e:
logger.error(f"Chat failed: {e}")
return {"success": False, "error": str(e), "response": None}