From b2e47e75a231baf5a7beca476256ab3d94e76c46 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 01 四月 2025 10:08:15 +0800
Subject: [PATCH] code加密

---
 app/service/difyService.py |    9 ++++-----
 1 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/app/service/difyService.py b/app/service/difyService.py
index 97de46b..595ca4c 100644
--- a/app/service/difyService.py
+++ b/app/service/difyService.py
@@ -72,9 +72,9 @@
             # print(response.text)
             return self._handle_response(response)
 
-    async def login(self, username: str, password: str, remember_me=True, invite_token:str="") -> str:
+    async def login(self, username: str, password: str, remember_me=True, invite_token:str="", email="") -> str:
         # password = RagflowCrypto(settings.PUBLIC_KEY, settings.PRIVATE_KEY).encrypt(password)
-        data = {"email": f"{username}@basic.com", "password": password, "remember_me": remember_me, "invite_token": invite_token,
+        data = {"email":username if "@" in username else f"{username}@basic.com", "password": password, "remember_me": remember_me, "invite_token": invite_token,
          "language": "zh-Hans"}
 
         async with httpx.AsyncClient() as client:
@@ -127,11 +127,10 @@
 
 
 
-    async def chat(self, token: str, user_id: int,  message: str, files: [], conversation_id: str):
-
+    async def chat(self, token: str, user_id: int,  message: str, files: [], conversation_id: str, inputs: dict):
         target_url = f"{self.base_url}/v1/chat-messages"
         data = {
-            "inputs": {},
+            "inputs": inputs,
             "query": message,
             "response_mode": "streaming",
             "conversation_id": conversation_id,

--
Gitblit v1.8.0