Merge pull request #267 from PySpur-Dev/fix/ollama-models

v0.1.17

Author: srijanpatel

backend/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pyspur"
-version = "0.1.16"
+version = "0.1.17"
 description = "PySpur is a Graph UI for building AI Agents in Python"
 requires-python = ">=3.11"
 license = "Apache-2.0"

backend/pyspur/nodes/llm/_utils.py

@@ -558,21 +558,13 @@ async def ollama_with_backoff(
 
     """
     client = AsyncClient(host=api_base)
-    print("=== Ollama Configuration ===")
-    print(f"Model: '{model}' '{model.replace('ollama/', '')}'")
-    print(f"API Base: {api_base}")
-    print(f"Messages: {messages}")
-    print(f"Format: {format}")
-    print(f"Options: {options}")
     try:
         response = await client.chat(
             model=model.replace("ollama/", ""),
             messages=messages,
             format=format,
             options=(options or OllamaOptions()).to_dict(),
         )
-        print("=== Ollama Response ===")
-        print(f"Response: {response}")
         return response.message.content
     except Exception as e:
         logging.error(f"Error calling Ollama API: {e}")