add: 사용자 채팅에 메타 수퍼바이저 추가 (#141)

* add: 메타 수퍼바이저 파이프라인 추가
closes #140

* add: 메타 수퍼바이저 파이프라인 추가
closes #140

Author: e7217

docker-compose.prod.yaml

@@ -33,6 +33,7 @@ services:
       - ./pipelines:/app/pipelines
     environment:
       - APISERVER_HOST=backend-ma:8000
+      - META_SUPERVISOR_HOST=meta-supervisor:8000
       - PIPELINES_DIR=/app/pipelines
       # 필요한 경우 특정 파이프라인 URL 추가 (주석 해제 후 사용)
       # - PIPELINES_URLS=https://github.com/open-webui/pipelines/blob/main/examples/filters/detoxify_filter_pipeline.py

pipelines/meta_supervisor_pipeline.py

@@ -0,0 +1,63 @@
+import os
+from typing import List, Union, Generator, Iterator
+from pydantic import BaseModel
+import requests
+
+APISERVER_HOST = os.getenv("META_SUPERVISOR_HOST")
+
+
+class RequestBody(BaseModel):
+    query: str
+    model: str
+    temperature: float
+
+
+class ResponseBody(BaseModel):
+    answer: str
+
+
+class Pipeline:
+    class Valves(BaseModel):
+        pass
+
+    def __init__(self):
+        self.name = "Meta Supervisor - Main"
+        self.agent_name = "metasupervisor"
+        self.endpoint = f"http://{APISERVER_HOST}/api/query"
+        pass
+
+    async def on_startup(self):
+        # This function is called when the server is started.
+        print(f"on_startup:{__name__}")
+        pass
+
+    async def on_shutdown(self):
+        # This function is called when the server is stopped.
+        print(f"on_shutdown:{__name__}")
+        pass
+
+    def pipe(
+        self, user_message: str, model_id: str, messages: List[dict], body: dict
+    ) -> Union[str, Generator, Iterator]:
+        # This is where you can add your custom pipelines like RAG.
+        print(f"pipe:{__name__}")
+
+        print("messages: ", messages)
+        print("user_message: ", user_message)
+        print("body: ", body)
+
+        headers = {}
+        headers["accept"] = "application/json"
+        headers["Content-Type"] = "application/json"
+
+        try:
+            r = requests.post(
+                url=f"{self.endpoint}",
+                json={"query": user_message },
+                headers=headers,
+            )
+
+            r.raise_for_status()
+            return ResponseBody(**r.json()).answer
+        except Exception as e:
+            return f"Error: {e}"