getting started on the preparedness data

This commit is contained in:
itsamejms
2025-09-13 17:27:46 +02:00
parent a86232a039
commit 8a2e7fdb6e
6 changed files with 341 additions and 1 deletions
+24
View File
@@ -0,0 +1,24 @@
import requests
def get_ollama_summary(text, model="gpt-oss:20b"):
url = "http://localhost:11434/v1/chat/completions"
payload = {
"model": model,
"messages": [
{"role": "system", "content": "You are a helpful assistant that summarizes text."},
{"role": "user", "content": text}
]
}
response = requests.post(url, json=payload)
response.raise_for_status()
return response.json()["choices"][0]["message"]["content"]
def get_ollama_embedding(text, model="nomic-embed-text"):
url = "http://localhost:11434/api/embeddings"
payload = {
"model": model,
"prompt": text
}
response = requests.post(url, json=payload)
response.raise_for_status()
return response.json()["embedding"]