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

update base

Browse Source
This commit is contained in:
Brian Pham
2023-12-12 23:56:20 +08:00
parent b7225cc674
commit ca1ccd7b91
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pr_agent/algo/ai_handlers/base_ai_handler.py
View File

@ -2,7 +2,7 @@ from abc import ABC, abstractmethod
class BaseAiHandler(ABC): class BaseAiHandler(ABC):
""" """
This class defines the interface for an AI handler. This class defines the interface for an AI handler to be used by the PR Agents.
""" """
@abstractmethod @abstractmethod
@ -15,6 +15,14 @@ class BaseAiHandler(ABC):
pass pass
@abstractmethod @abstractmethod
"""
This method should be implemented to return a chat completion from the AI model.
params:
model: the name of the model to use for the chat completion
system: the system message string to use for the chat completion
user: the user message string to use for the chat completion
temperature: the temperature to use for the chat completion
"""
async def chat_completion(self, model: str, system: str, user: str, temperature: float = 0.2): async def chat_completion(self, model: str, system: str, user: str, temperature: float = 0.2):
pass pass