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

Logging

Browse Source
This commit is contained in:
zmeir
2023-08-07 22:42:53 +03:00
parent 6c4a5bae52
commit 7f1849a867
2 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pr_agent/algo/ai_handler.py
View File

@ -74,9 +74,15 @@ class AiHandler:
TryAgain: If there is an attribute error during OpenAI inference. TryAgain: If there is an attribute error during OpenAI inference.
""" """
try: try:
deployment_id = self.deployment_id
if get_settings().config.verbosity_level >= 2:
logging.debug(
f"Generating completion with {model}"
f"{(' from deployment ' + deployment_id) if deployment_id else ''}"
)
response = await acompletion( response = await acompletion(
model=model, model=model,
deployment_id=self.deployment_id, deployment_id=deployment_id,
messages=[ messages=[
{"role": "system", "content": system}, {"role": "system", "content": system},
{"role": "user", "content": user} {"role": "user", "content": user}

7
pr_agent/algo/pr_processing.py
View File

@ -214,7 +214,6 @@ async def retry_with_fallback_models(f: Callable):
if not isinstance(fallback_models, list): if not isinstance(fallback_models, list):
fallback_models = [m.strip() for m in fallback_models.split(",")] fallback_models = [m.strip() for m in fallback_models.split(",")]
all_models = [model] + fallback_models all_models = [model] + fallback_models
# getting all deployments # getting all deployments
deployment_id = get_settings().get("openai.deployment_id", None) deployment_id = get_settings().get("openai.deployment_id", None)
fallback_deployments = get_settings().get("openai.fallback_deployments", []) fallback_deployments = get_settings().get("openai.fallback_deployments", [])
@ -230,7 +229,11 @@ async def retry_with_fallback_models(f: Callable):
get_settings().set("openai.deployment_id", deployment_id) get_settings().set("openai.deployment_id", deployment_id)
return await f(model) return await f(model)
except Exception as e: except Exception as e:
logging.warning(f"Failed to generate prediction with {model}: {traceback.format_exc()}") logging.warning(
f"Failed to generate prediction with {model}"
f"{(' from deployment ' + deployment_id) if deployment_id else ''}: "
f"{traceback.format_exc()}"
)
if i == len(all_models) - 1: # If it's the last iteration if i == len(all_models) - 1: # If it's the last iteration
raise # Re-raise the last exception raise # Re-raise the last exception