xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-03 04:10:49 +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):
"""
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
@ -15,6 +15,14 @@ class BaseAiHandler(ABC):
pass
@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):
pass