mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-23 07:54:27 +01:00
27 lines
864 B
Python
27 lines
864 B
Python
import importlib
|
|
from typing import Optional
|
|
from evals.api import CompletionFn, CompletionResult
|
|
|
|
from evals.prompt.base import CompletionPrompt
|
|
from evals.record import record_sampling
|
|
from auto_gpt_benchmarking.AutoGPTAgent import AutoGPTAgent
|
|
|
|
|
|
class AutoGPTCompletionResult(CompletionResult):
|
|
def __init__(self, response) -> None:
|
|
self.response = response
|
|
|
|
def get_completions(self) -> list[str]:
|
|
return [self.response.strip()]
|
|
|
|
|
|
class AutoGPTCompletionFn(CompletionFn):
|
|
def __init__(self, **kwargs) -> None:
|
|
pass
|
|
|
|
def __call__(self, prompt, **kwargs) -> AutoGPTCompletionResult:
|
|
prompt = CompletionPrompt(prompt).to_formatted_prompt()
|
|
agent = AutoGPTAgent(prompt)
|
|
response = agent.start()
|
|
record_sampling(prompt=prompt, sampled=response)
|
|
return AutoGPTCompletionResult(response) |