xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-02 20:00:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

artifact

Browse Source
This commit is contained in:
mrT23
2024-02-25 10:45:15 +02:00
parent 984a2888ae
commit dad3d3429f
8 changed files with 22 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/algo/ai_handlers/litellm_ai_handler.py
View File

@ -115,7 +115,7 @@ class LiteLLMAIHandler(BaseAiHandler):
if self.aws_bedrock_client:
kwargs["aws_bedrock_client"] = self.aws_bedrock_client
get_logger().debug("Prompts", messages={"system": system, "user": user})
get_logger().debug("Prompts", artifact={"system": system, "user": user})
response = await acompletion(**kwargs)
except (APIError, Timeout, TryAgain) as e:
get_logger().error("Error during OpenAI inference: ", e)
@ -133,6 +133,6 @@ class LiteLLMAIHandler(BaseAiHandler):
finish_reason = response["choices"][0]["finish_reason"]
# usage = response.get("usage")
get_logger().debug(f"\nAI response:\n{resp}")
get_logger().debug("full_response", response=response)
get_logger().debug("Full_response", artifact=response)
return resp, finish_reason