Ollama Support

[sprajosh]

📥 Pull Request

📘 Description Add support for Ollama support by patching the ollama.chat function.

🔄 Related Issue (if applicable) Ollama support #192

🎯 Goal Add support for official Ollama python library.

🔍 Additional Context Any extra information or context to help us understand the change?

🧪 Testing

import ollama
import agentops

AGENTOPS_API_KEY = "<api-key>"
agentops.init(AGENTOPS_API_KEY)

import ollama

# Sync chat
response = ollama.chat(
    model='orca-mini',
    messages=[{'role': 'user', 'content': 'Why is the sky blue?'}
])

# Async chat
response = ollama.chat(
    model="orca-mini",
    messages=[{"role": "user", "content": "Why is the sky blue?"}],
    stream=True,
)

for chunk in response:
    print(chunk)

agentops.end_session("Success")

This is a first draft. I'd like some feedback to understand if I'm missing something. ~Also, I don't see the analytics on session drill-down view. I will have to check the frontend project as well to see if this is happening because ollama is an unknown event.~

Todo

• [x] Ollama sync
• [x] Ollama sync with stream
• [x] Ollama sync client
• [x] Ollama async client
• [x] Count input and output tokens
• [x] Undo override for all
• [x] Minimum ollama version support

Dependencies ~https://github.com/AgentOps-AI/tokencost/pull/49 - Ollama support in tokencost to count token from message~ Token cost is calculated on server.

[siyangqiu]

It looks like ollama has a streaming mode. I don't know if you want to add support for that

[sprajosh]

It looks like ollama has a streaming mode. I don't know if you want to add support for that

Yes, have added support for ollama.chat, ollama.chat with stream, ollama.Client.chat, ollama.AsyncClient.chat.

[siyangqiu]

Awesome! Thanks for making the changes. Sorry I wasn't clear enough about token cost! I'll test this again if you can remove tokencost

[gitguardian[bot]]

️✅ There are no secrets present in this pull request anymore.

If these secrets were true positive and are still valid, we highly recommend you to revoke them. Once a secret has been leaked into a git repository, you should consider it compromised, even if it was deleted immediately. Find here more information about risks.

---

^{_{🦉 GitGuardian detects secrets in your source code to help developers and security teams secure the modern development process. You are seeing this because you or someone else with access to this repository has authorized GitGuardian to scan your pull request.}}

[siyangqiu]

I just tested this and it works! Good work! 🎉

I noticed that prompt tokens weren't being counted, but I suspect that's something I need to fix on the API server.