wip

Author: nerdai

src/llm_agents_from_scratch/llms/ollama.py

@@ -37,7 +37,11 @@ async def complete(self, prompt: str, **kwargs: Any) -> CompleteResult:
         Returns:
             CompleteResult: The text completion result.
         """
-        response = await self._client.generate(model=self.model, prompt=prompt)
+        response = await self._client.generate(
+            model=self.model,
+            prompt=prompt,
+            **kwargs,
+        )
         return CompleteResult(
             response=response.response,
             prompt=prompt,