chore: import storyforge baseline clean

collector-service/app/fastgpt.py

@@ -0,0 +1,48 @@
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+
+class FastGPTClient:
+    def __init__(self, *, base_url: str, dataset_api_key: str, timeout: float = 60.0) -> None:
+        self.base_url = base_url.rstrip("/")
+        self.dataset_api_key = dataset_api_key.strip()
+        self.timeout = timeout
+
+    @property
+    def enabled(self) -> bool:
+        return bool(self.base_url and self.dataset_api_key)
+
+    async def ensure_dataset(self, name: str, intro: str = "") -> dict[str, Any] | None:
+        if not self.enabled:
+            return None
+        payload = {"name": name, "intro": intro}
+        async with httpx.AsyncClient(timeout=self.timeout) as client:
+            response = await client.post(
+                f"{self.base_url}/api/core/dataset/create",
+                headers={"Authorization": f"Bearer {self.dataset_api_key}"},
+                json=payload,
+            )
+            response.raise_for_status()
+            return response.json().get("data") or response.json()
+
+    async def add_text_document(self, dataset_id: str, name: str, text: str) -> dict[str, Any] | None:
+        if not self.enabled or not dataset_id.strip():
+            return None
+        payload = {
+            "datasetId": dataset_id,
+            "type": "text",
+            "name": name,
+            "trainingType": "chunk",
+            "text": text,
+        }
+        async with httpx.AsyncClient(timeout=self.timeout) as client:
+            response = await client.post(
+                f"{self.base_url}/api/core/dataset/collection/create/text",
+                headers={"Authorization": f"Bearer {self.dataset_api_key}"},
+                json=payload,
+            )
+            response.raise_for_status()
+            return response.json().get("data") or response.json()