Add sdk for Agent API (#3220)

### What problem does this PR solve? Add sdk for Agent API ### Type of change - [x] New Feature (non-breaking change which adds functionality) --------- Co-authored-by: liuhua <10215101452@stu.ecun.edu.cn>
2025-12-08 20:42:30 +08:00 · 2024-11-06 18:03:45 +08:00
parent 0dff64f6ad
commit f3aaa0d453
7 changed files with 546 additions and 92 deletions
--- a/sdk/python/ragflow_sdk/modules/agent.py
+++ b/sdk/python/ragflow_sdk/modules/agent.py
@ -0,0 +1,59 @@
+from .base import Base
+from .session import Session
+import requests
+
+class Agent(Base):
+    def __init__(self,rag,res_dict):
+        self.id  = None
+        self.avatar = None
+        self.canvas_type = None
+        self.description = None
+        self.dsl = None
+        super().__init__(rag, res_dict)
+
+    class Dsl(Base):
+        def __init__(self,rag,res_dict):
+            self.answer = []
+            self.components = {
+                "begin": {
+                    "downstream": ["Answer:China"],
+                    "obj": {
+                        "component_name": "Begin",
+                        "params": {}
+                    },
+                    "upstream": []
+                }
+            }
+            self.graph = {
+                "edges": [],
+                "nodes": [
+                    {
+                        "data": {
+                            "label": "Begin",
+                            "name": "begin"
+                        },
+                        "id": "begin",
+                        "position": {
+                            "x": 50,
+                            "y": 200
+                        },
+                        "sourcePosition": "left",
+                        "targetPosition": "right",
+                        "type": "beginNode"
+                    }
+                ]
+            }
+            self.history =  []
+            self.messages =  []
+            self.path =  []
+            self.reference = []
+            super().__init__(rag,res_dict)
+
+    @staticmethod
+    def create_session(id,rag) -> Session:
+        res = requests.post(f"http://127.0.0.1:9380/api/v1/agents/{id}/sessions",headers={"Authorization": f"Bearer {rag.user_key}"},json={})
+        res = res.json()
+        if res.get("code") == 0:
+            return Session(rag,res.get("data"))
+        raise Exception(res.get("message"))
+