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

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
Show stats 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