PSE2_FF
/
pse2_ff
0
0
Fork 0
Code Issues Pull Requests Packages Projects 1 Releases Wiki Activity

Dynamischer Prompt mit API-KPIs, Fortschritt integriert

pull/96/head
Abdulrahman Dabbagh 2025-06-25 19:28:19 +02:00
parent 4ca8314ed2
commit 360d7f4906
1 changed files with 33 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
project/backend/exxetaGPT-service/extractExxeta.py
View File

@ -1,7 +1,6 @@
import requests import requests
import json import json
import os import os
import time
import logging import logging
from dotenv import load_dotenv from dotenv import load_dotenv
@ -17,6 +16,18 @@ TIMEOUT = 180
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
def get_dynamic_labels():
url = f"{COORDINATOR_URL}/api/kpi_setting/"
try:
response = requests.get(url, timeout=10)
response.raise_for_status()
kpi_list = response.json()
labels = [kpi["name"].upper() for kpi in kpi_list if kpi.get("active", False)]
return labels
except Exception as e:
logger.warning(f"Konnte dynamische Labels nicht laden: {e}")
return []
def extract_with_exxeta(pages_json, pitchbook_id): def extract_with_exxeta(pages_json, pitchbook_id):
results = [] results = []
@ -30,9 +41,7 @@ def extract_with_exxeta(pages_json, pitchbook_id):
if i % 8 == 0: if i % 8 == 0:
requests.post(COORDINATOR_URL + "/api/progress", json={"id": pitchbook_id, "progress": 35 + 60/len(pages_json)*i}) requests.post(COORDINATOR_URL + "/api/progress", json={"id": pitchbook_id, "progress": 35 + 60/len(pages_json)*i})
page_num = page_data.get("page") page_num = page_data.get("page")
page_data.get("page")
text = page_data.get("text", "") text = page_data.get("text", "")
if not text: if not text:
@ -51,9 +60,9 @@ def extract_with_exxeta(pages_json, pitchbook_id):
"- Gib die Antwort als **JSON-Array** im folgenden Format zurück:\n\n" "- Gib die Antwort als **JSON-Array** im folgenden Format zurück:\n\n"
"[\n" "[\n"
" {\n" " {\n"
" \"label\": \"FONDSNAME\",\n" ' "label": "FONDSNAME",\n'
" \"entity\": \"...\",\n" ' "entity": "...",\n'
f" \"page\": {page_num},\n" f' "page": {page_num},\n'
" },\n" " },\n"
" ...\n" " ...\n"
"]\n\n" "]\n\n"
@ -61,45 +70,29 @@ def extract_with_exxeta(pages_json, pitchbook_id):
f"TEXT:\n{text}" f"TEXT:\n{text}"
) )
else: else:
labels = get_dynamic_labels()
prompt_kennzahlen = "".join([f"- {label}\n" for label in labels])
prompt = ( prompt = (
"Bitte extrahiere relevante Fondskennzahlen aus dem folgenden Pitchbook-Text. " "Bitte extrahiere relevante Fondskennzahlen aus dem folgenden Pitchbook-Text. "
"Analysiere den Text sorgfältig, um **nur exakt benannte und relevante Werte** zu extrahieren.\n\n" "Analysiere den Text sorgfältig, um **nur exakt benannte und relevante Werte** zu extrahieren.\n\n"
"ZU EXTRAHIERENDE KENNZAHLEN (immer exakt wie unten angegeben):\n" "ZU EXTRAHIERENDE KENNZAHLEN (immer exakt wie unten angegeben):\n"
"- FONDSNAME\n" f"{prompt_kennzahlen}\n"
"- FONDSMANAGER\n"
"- AIFM (z. B. Name Kapitalverwaltungsgesellschaft)\n"
"- DATUM\n"
"- RISIKOPROFIL (z. B. CORE, CORE+, VALUE-ADDED, OPPORTUNISTISCH)\n"
"- ARTIKEL (z. B. ARTIKEL 6, 8, 9)\n"
"- ZIELRENDITE\n"
"- RENDITE\n"
"- ZIELAUSSCHÜTTUNG\n"
"- AUSSCHÜTTUNG\n"
"- LAUFZEIT\n"
"- LTV\n"
"- MANAGEMENTGEBÜHREN (ggf. mit Staffelung und Bezug auf NAV/GAV)\n"
"- SEKTORENALLOKATION (z. B. BÜRO, LOGISTIK, WOHNEN... inkl. %-Angaben)\n"
"- LÄNDERALLOKATION (z. B. DEUTSCHLAND, FRANKREICH, etc. inkl. %-Angaben)\n\n"
"WICHTIG:\n" "WICHTIG:\n"
"- Gib **nur eine Entität pro Kennzahl** an - keine Listen oder Interpretationen.\n" "- Gib **nur eine Entität pro Kennzahl** an - keine Listen oder Interpretationen.\n"
"- Wenn mehrere Varianten genannt werden (z. B. \"Core und Core+\"), gib sie im Originalformat als **eine entity** an.\n" '- Wenn mehrere Varianten genannt werden (z. B. "Core und Core+"), gib sie im Originalformat als **eine entity** an.\n'
"- **Keine Vermutungen oder Ergänzungen**. Wenn keine Information enthalten ist, gib die Kennzahl **nicht aus**.\n" "- **Keine Vermutungen oder Ergänzungen**. Wenn keine Information enthalten ist, gib die Kennzahl **nicht aus**.\n"
"- Extrahiere **nur wörtlich vorkommende Inhalte** (keine Berechnungen, keine Zusammenfassungen).\n" "- Extrahiere **nur wörtlich vorkommende Inhalte** (keine Berechnungen, keine Zusammenfassungen).\n"
"- Jeder gefundene Wert muss einem der obigen Label **eindeutig zuordenbar** sein.\n\n" "- Jeder gefundene Wert muss einem der obigen Label **eindeutig zuordenbar** sein.\n\n"
"FORMAT:\n" "FORMAT:\n"
"Antworte als **reines JSON-Array** mit folgendem Format:\n" "Antworte als **reines JSON-Array** mit folgendem Format:\n"
"[\n" "[\n"
" {\n" " {\n"
" \"label\": \"Kennzahlname (exakt wie oben)\",\n" ' "label": "Kennzahlname (exakt wie oben)",\n'
" \"entity\": \"Wert aus dem Text (exakt im Original)\",\n" ' "entity": "Wert aus dem Text (exakt im Original)",\n'
f" \"page\": {page_num},\n" f' "page": {page_num},\n'
" },\n" " },\n"
" ...\n" " ...\n"
"]\n\n" "]\n\n"
f"Falls keine Kennzahl enthalten ist, gib ein leeres Array [] zurück.\n\n" f"Falls keine Kennzahl enthalten ist, gib ein leeres Array [] zurück.\n\n"
f"Nur JSON-Antwort - keine Kommentare, keine Erklärungen, kein Text außerhalb des JSON.\n\n" f"Nur JSON-Antwort - keine Kommentare, keine Erklärungen, kein Text außerhalb des JSON.\n\n"
f"TEXT:\n{text}" f"TEXT:\n{text}"
@ -119,16 +112,17 @@ def extract_with_exxeta(pages_json, pitchbook_id):
"temperature": 0.0 "temperature": 0.0
} }
print("\n==== Dynamisch gebauter Prompt ====\n")
print(prompt)
print("\n===================================\n")
url = f"{EXXETA_BASE_URL}/deployments/{MODEL}/chat/completions" url = f"{EXXETA_BASE_URL}/deployments/{MODEL}/chat/completions"
for attempt in range(1, MAX_RETRIES + 1): for attempt in range(1, MAX_RETRIES + 1):
try: try:
response = requests.post(url, headers=headers, json=payload, timeout=TIMEOUT) response = requests.post(url, headers=headers, json=payload, timeout=TIMEOUT)
response.raise_for_status() response.raise_for_status()
content = response.json()["choices"][0]["message"]["content"].strip()
content = response.json()["choices"][0]["message"]["content"]
content = content.strip()
if content.startswith("```json"): if content.startswith("```json"):
content = content.split("```json")[1] content = content.split("```json")[1]
if content.endswith("```"): if content.endswith("```"):
@ -143,14 +137,16 @@ def extract_with_exxeta(pages_json, pitchbook_id):
if isinstance(page_results, list): if isinstance(page_results, list):
results.extend(page_results) results.extend(page_results)
break break
except requests.exceptions.RequestException:
except requests.exceptions.RequestException as e:
if attempt == MAX_RETRIES: if attempt == MAX_RETRIES:
results.extend([]) results.extend([])
except Exception as e: except Exception:
if attempt == MAX_RETRIES: if attempt == MAX_RETRIES:
results.extend([]) results.extend([])
requests.post(COORDINATOR_URL + "/api/progress", json={"id": pitchbook_id, "progress": 95}) requests.post(COORDINATOR_URL + "/api/progress", json={"id": pitchbook_id, "progress": 95})
return json.dumps(results, indent=2, ensure_ascii=False) return json.dumps(results, indent=2, ensure_ascii=False)
if __name__ == "__main__":
print("📡 Test-Aufruf get_dynamic_labels:")
print(get_dynamic_labels())