mirror of
https://github.com/jooray/rss2podcast.git
synced 2025-05-23 07:52:00 +00:00
14 lines
431 B
Python
14 lines
431 B
Python
|
import ollama
|
||
|
|
|
||
|
|
class OllamaClient:
|
||
|
|
def __init__(self, model, url='http://localhost:11434'):
|
||
|
|
self.model = model
|
||
|
|
self.client = ollama.Client(host=url)
|
||
|
|
|
||
|
|
def generate(self, prompt):
|
||
|
|
try:
|
||
|
|
response = self.client.generate(model=self.model, prompt=prompt)
|
||
|
|
return response['response']
|
||
|
|
except Exception as e:
|
||
|
|
raise Exception(f"Ollama API request failed: {str(e)}")
|