Skip to content

Commit f9bd40c

Browse files
committed
rename files to prevent bad imports
1 parent ec30658 commit f9bd40c

8 files changed

Lines changed: 4 additions & 4 deletions

File tree

File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.

agentops/llms/providers/ollama.py renamed to agentops/llms/providers/ollama_provider.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

File renamed without changes.

0 commit comments

Comments
 (0)