highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-10-10 13:15:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add litellm.set_verbose support

Browse Source
This commit is contained in:
Weves
2024-07-02 23:01:05 -07:00
committed by Chris Weaver
parent 2db128fb36
commit 982b1b0c49
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/danswer/llm/chat_llm.py
View File

@@ -42,6 +42,8 @@ logger = setup_logger()
litellm.drop_params = True litellm.drop_params = True
litellm.telemetry = False litellm.telemetry = False
litellm.set_verbose = LOG_ALL_MODEL_INTERACTIONS
def _base_msg_to_role(msg: BaseMessage) -> str: def _base_msg_to_role(msg: BaseMessage) -> str:
if isinstance(msg, HumanMessage) or isinstance(msg, HumanMessageChunk): if isinstance(msg, HumanMessage) or isinstance(msg, HumanMessageChunk):