From c4586d37f5171d55de75138ce969650ff84e921f Mon Sep 17 00:00:00 2001 From: yyhhyyyyyy <95077259+yyhhyyyyyy@users.noreply.github.com> Date: Thu, 12 Dec 2024 14:32:17 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20style:=20format=20llm.py=20code?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/services/llm.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/app/services/llm.py b/app/services/llm.py index 5b3baa7..2c45ef9 100644 --- a/app/services/llm.py +++ b/app/services/llm.py @@ -179,7 +179,10 @@ def _generate_response(prompt: str) -> str: headers={"Authorization": f"Bearer {api_key}"}, json={ "messages": [ - {"role": "system", "content": "You are a friendly assistant"}, + { + "role": "system", + "content": "You are a friendly assistant", + }, {"role": "user", "content": prompt}, ] }, @@ -197,7 +200,9 @@ def _generate_response(prompt: str) -> str: "client_secret": secret_key, } access_token = ( - requests.post("https://aip.baidubce.com/oauth/2.0/token", params=params) + requests.post( + "https://aip.baidubce.com/oauth/2.0/token", params=params + ) .json() .get("access_token") )