github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 15:19:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: Fix sparkllm authentication problem. (#18651)

Browse Source 
- **Description:** fix sparkllm authentication problem.The current
timestamp is in RFC1123 format. The time deviation must be controlled
within 300s. I changed to re-obtain the url every time I ask a question.
https://www.xfyun.cn/doc/spark/general_url_authentication.html#_1-2-%E9%89%B4%E6%9D%83%E5%8F%82%E6%95%B0
This commit is contained in:
Guangdong Liu 2024-03-07 10:43:16 +08:00 committed by GitHub
parent 89d32ffbbd
commit ced5e7bae7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libs/community/langchain_community/chat_models/sparkllm.py
View File

@ -306,15 +306,12 @@ class _SparkLLMClient:
"wss://spark-api.xf-yun.com/v3.1/chat" if not api_url else api_url "wss://spark-api.xf-yun.com/v3.1/chat" if not api_url else api_url
) )
self.app_id = app_id self.app_id = app_id
self.ws_url = _SparkLLMClient._create_url(
self.api_url,
api_key,
api_secret,
)
self.model_kwargs = model_kwargs self.model_kwargs = model_kwargs
self.spark_domain = spark_domain or "generalv3" self.spark_domain = spark_domain or "generalv3"
self.queue: Queue[Dict] = Queue() self.queue: Queue[Dict] = Queue()
self.blocking_message = {"content": "", "role": "assistant"} self.blocking_message = {"content": "", "role": "assistant"}
self.api_key = api_key
self.api_secret = api_secret
@staticmethod @staticmethod
def _create_url(api_url: str, api_key: str, api_secret: str) -> str: def _create_url(api_url: str, api_key: str, api_secret: str) -> str:
@ -370,7 +367,11 @@ class _SparkLLMClient:
) -> None: ) -> None:
self.websocket_client.enableTrace(False) self.websocket_client.enableTrace(False)
ws = self.websocket_client.WebSocketApp( ws = self.websocket_client.WebSocketApp(
self.ws_url, _SparkLLMClient._create_url(
self.api_url,
self.api_key,
self.api_secret,
),
on_message=self.on_message, on_message=self.on_message,
on_error=self.on_error, on_error=self.on_error,
on_close=self.on_close, on_close=self.on_close,

13
libs/community/langchain_community/llms/sparkllm.py
View File

@ -203,15 +203,12 @@ class _SparkLLMClient:
"wss://spark-api.xf-yun.com/v3.1/chat" if not api_url else api_url "wss://spark-api.xf-yun.com/v3.1/chat" if not api_url else api_url
) )
self.app_id = app_id self.app_id = app_id
self.ws_url = _SparkLLMClient._create_url(
self.api_url,
api_key,
api_secret,
)
self.model_kwargs = model_kwargs self.model_kwargs = model_kwargs
self.spark_domain = spark_domain or "generalv3" self.spark_domain = spark_domain or "generalv3"
self.queue: Queue[Dict] = Queue() self.queue: Queue[Dict] = Queue()
self.blocking_message = {"content": "", "role": "assistant"} self.blocking_message = {"content": "", "role": "assistant"}
self.api_key = api_key
self.api_secret = api_secret
@staticmethod @staticmethod
def _create_url(api_url: str, api_key: str, api_secret: str) -> str: def _create_url(api_url: str, api_key: str, api_secret: str) -> str:
@ -267,7 +264,11 @@ class _SparkLLMClient:
) -> None: ) -> None:
self.websocket_client.enableTrace(False) self.websocket_client.enableTrace(False)
ws = self.websocket_client.WebSocketApp( ws = self.websocket_client.WebSocketApp(
self.ws_url, _SparkLLMClient._create_url(
self.api_url,
self.api_key,
self.api_secret,
),
on_message=self.on_message, on_message=self.on_message,
on_error=self.on_error, on_error=self.on_error,
on_close=self.on_close, on_close=self.on_close,