From ddca70d174bf390e4ceca0c2c2a4fb2334d7c17e Mon Sep 17 00:00:00 2001
From: Renato Rosa <renatoxsr@gmail.com>
Date: Wed, 19 Aug 2026 19:53:26 -0300
Subject: [PATCH] Adicionar app/services/ollama_service.py

---
 app/services/ollama_service.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 app/services/ollama_service.py

diff --git a/app/services/ollama_service.py b/app/services/ollama_service.py
new file mode 100644
index 0000000..88dc558
--- /dev/null
+++ b/app/services/ollama_service.py
@@ -0,0 +1,24 @@
+import requests
+import json
+
+config = {}
+config["gen_host"] = os.environ.get("GEN_HOST","")
+
+if not config["gen_host"]:
+    raise ValueError("Generative backend invalid")
+
+class OllamaService:
+    def __init__(self):
+        self.url = f"http://{config["gen_host"]}:11434/api/generate"
+
+    def generate(self, prompt: str):
+        payload = {
+            "model": "llama3.1:8b",
+            "prompt": prompt,
+            "stream": False
+        }
+
+        response = requests.post(self.url, json=payload)
+        data = response.json()
+
+        return data.get("response", "")