From 498ea113e09d7781a79ca67078de6bbc13bdf2f6 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 08 四月 2025 09:56:42 +0800
Subject: [PATCH] license同步任务

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

diff --git a/app/service/difyService.py b/app/service/difyService.py
index 075cfaa..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:
@@ -128,7 +128,6 @@
 
 
     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,

--
Gitblit v1.8.0