ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-14 20:05:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

adding support for Anthropic, Cohere, Replicate, Azure

Browse source
This commit is contained in:
Krish Dholakia 2023-08-03 12:04:08 -07:00 committed by GitHub
parent 26b008565b
commit 102edcdcf1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pr_agent/algo/ai_handler.py
View file

@ -3,7 +3,7 @@ import logging
import openai import openai
from openai.error import APIError, RateLimitError, Timeout, TryAgain from openai.error import APIError, RateLimitError, Timeout, TryAgain
from retry import retry from retry import retry
from litellm import acompletion
from pr_agent.config_loader import get_settings from pr_agent.config_loader import get_settings
OPENAI_RETRIES=5 OPENAI_RETRIES=5
@ -57,7 +57,7 @@ class AiHandler:
TryAgain: If there is an attribute error during OpenAI inference. TryAgain: If there is an attribute error during OpenAI inference.
""" """
try: try:
response = await openai.ChatCompletion.acreate( response = await acompletion(
model=model, model=model,
deployment_id=self.deployment_id, deployment_id=self.deployment_id,
messages=[ messages=[