@@ -64,7 +64,7 @@ def override(self):
6464
6565 def undo_override (self ):
6666 if original_func is not None and original_func != {}:
67- import agentops . llms . providers . ollama as ollama
67+ import ollama
6868
6969 ollama .chat = original_func ["ollama.chat" ]
7070 ollama .Client .chat = original_func ["ollama.Client.chat" ]
@@ -74,7 +74,7 @@ def __init__(self, client):
7474 super ().__init__ (client )
7575
7676 def _override_chat (self ):
77- import agentops . llms . providers . ollama as ollama
77+ import ollama
7878
7979 original_func ["ollama.chat" ] = ollama .chat
8080
@@ -88,7 +88,7 @@ def patched_function(*args, **kwargs):
8888 ollama .chat = patched_function
8989
9090 def _override_chat_client (self ):
91- from agentops . llms . providers . ollama import Client
91+ from ollama import Client
9292
9393 original_func ["ollama.Client.chat" ] = Client .chat
9494
@@ -102,7 +102,7 @@ def patched_function(*args, **kwargs):
102102 Client .chat = patched_function
103103
104104 def _override_chat_async_client (self ):
105- from agentops . llms . providers . ollama import AsyncClient
105+ from ollama import AsyncClient
106106
107107 original_func ["ollama.AsyncClient.chat" ] = AsyncClient .chat
108108
0 commit comments