ldooks/dragon-bot
ldooks/dragon-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add model name to llm output
All checks were successful
Build and push / changes (push) Successful in 39s
Details
Build and push / Lint-Python (push) Successful in 2s
Details
Build and push / Build-and-Push-Docker (push) Successful in 1m17s
Details
Build and push / post-status-to-discord (push) Successful in 1s
Details
Build and push / sync-argocd-app (push) Successful in 2s
Details

Browse Source
This commit is contained in:
Luke R 2025-03-13 14:33:24 -07:00
parent 1ce3452f8a
commit 55ab511784
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
app/core_utils.py
View File

@ -174,7 +174,10 @@ async def send_to_llm(ctx, message):
response = requests.post(url, json=payload, headers=headers) response = requests.post(url, json=payload, headers=headers)
answer = response.json()["message"]["content"] answer = response.json()["message"]["content"]
return remove_between(answer, "<think>", "</think>") return (
remove_between(answer, "<think>", "</think>")
+ f"\n-# Generated using {llm_rules.get('model')}"
)
except Exception as e: except Exception as e:
print(e) print(e)
return "Somethings wrong, maybe the LLM crashed" return "Somethings wrong, maybe the LLM crashed"