8ae1e64987
Co-authored-by: Charles Packer <packercharles@gmail.com> Co-authored-by: Shubham Naik <shubham.naik10@gmail.com> Co-authored-by: Shubham Naik <shub@memgpt.ai>
12 lines
408 B
Python
12 lines
408 B
Python
from abc import ABC, abstractmethod
|
|
|
|
|
|
class LLMChatCompletionWrapper(ABC):
|
|
@abstractmethod
|
|
def chat_completion_to_prompt(self, messages, functions, function_documentation=None):
|
|
"""Go from ChatCompletion to a single prompt string"""
|
|
|
|
@abstractmethod
|
|
def output_to_chat_completion_response(self, raw_llm_output):
|
|
"""Turn the LLM output string into a ChatCompletion response"""
|