From 2c1545366ad4936f4a4f4d4ed390976ee12e7e78 Mon Sep 17 00:00:00 2001
From: Parsa <spacenavard1@gmail.com>
Date: Sun, 14 Sep 2025 00:45:50 +0330
Subject: [PATCH 1/2] fix provider issue

---
 iragent/agent.py  | 2 +-
 iragent/models.py | 5 ++---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/iragent/agent.py b/iragent/agent.py
index 3354633..6c97c46 100644
--- a/iragent/agent.py
+++ b/iragent/agent.py
@@ -30,7 +30,7 @@ def __init__(
         max_token: int = 100,
         next_agent: str = None,
         fn: List[Callable] = [],
-        provider: str = "openai",
+        provider: str = None,
         response_format: str = None,
         memory = None
     ):
diff --git a/iragent/models.py b/iragent/models.py
index b436c3a..23fcb50 100644
--- a/iragent/models.py
+++ b/iragent/models.py
@@ -294,7 +294,7 @@ def __init__(
         api_key: str,
         temperature: float = 0.1,
         max_token: int = 512,
-        provider: str = "openai",
+        provider: str = None,
     ) -> None:
         self.chunk_size = chunk_size
         self.summerize_agent = Agent(
@@ -468,7 +468,7 @@ def _summarize_page(self, result, query: str):
         chunks = chunker(page_text, token_limit=self.chunk_size)
 
         # Ollama servers dislike shared clients; make one per thread
-        if getattr(self.summerize_agent, "provider", "").lower() == "ollama":
+        if getattr(self.summerize_agent, "provider", "").lower() == "ollama" or self.summerize_agent.provider == None:
             summarizer = Agent(
                 name="Summarize Agent (thread‑local)",
                 model=self.summerize_agent.model,
@@ -477,7 +477,6 @@ def _summarize_page(self, result, query: str):
                 system_prompt=self.summerize_agent.system_prompt,
                 temprature=self.summerize_agent.temprature,
                 max_token=self.summerize_agent.max_token,
-                provider="ollama",
             )
         else:
             summarizer = self.summerize_agent  # safe to reuse for OpenAI etc.

From 948a45666e948c88a2e4b057dedd4ca79b955844 Mon Sep 17 00:00:00 2001
From: Parsa <spacenavard1@gmail.com>
Date: Sun, 14 Sep 2025 00:48:43 +0330
Subject: [PATCH 2/2] fix ruff

---
 iragent/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/iragent/models.py b/iragent/models.py
index 23fcb50..be08b91 100644
--- a/iragent/models.py
+++ b/iragent/models.py
@@ -468,7 +468,7 @@ def _summarize_page(self, result, query: str):
         chunks = chunker(page_text, token_limit=self.chunk_size)
 
         # Ollama servers dislike shared clients; make one per thread
-        if getattr(self.summerize_agent, "provider", "").lower() == "ollama" or self.summerize_agent.provider == None:
+        if getattr(self.summerize_agent, "provider", "").lower() == "ollama" :
             summarizer = Agent(
                 name="Summarize Agent (thread‑local)",
                 model=self.summerize_agent.model,