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..be08b91 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" : 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.