injectlib/MoneyPrinterTurbo
1
0
Fork 0
mirror of https://github.com/harry0703/MoneyPrinterTurbo.git synced 2025-11-25 11:29:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

🎨 style: format llm.py code

Browse Source
This commit is contained in:
yyhhyyyyyy
2024-12-12 14:32:17 +08:00
parent 2d8cd23fe7
commit c4586d37f5
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/services/llm.py
View File

@@ -179,7 +179,10 @@ def _generate_response(prompt: str) -> str:
headers={"Authorization": f"Bearer {api_key}"}, headers={"Authorization": f"Bearer {api_key}"},
json={ json={
"messages": [ "messages": [
{"role": "system", "content": "You are a friendly assistant"}, {
"role": "system",
"content": "You are a friendly assistant",
},
{"role": "user", "content": prompt}, {"role": "user", "content": prompt},
] ]
}, },
@@ -197,7 +200,9 @@ def _generate_response(prompt: str) -> str:
"client_secret": secret_key, "client_secret": secret_key,
} }
access_token = ( 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() .json()
.get("access_token") .get("access_token")
) )