diff --git a/src/dispatch/plugins/bases/artificial_intelligence.py b/src/dispatch/plugins/bases/artificial_intelligence.py index cb185686b13a..247910be113e 100644 --- a/src/dispatch/plugins/bases/artificial_intelligence.py +++ b/src/dispatch/plugins/bases/artificial_intelligence.py @@ -5,17 +5,15 @@ :license: Apache, see LICENSE for more details. .. moduleauthor:: Marc Vilanova """ + from dispatch.plugins.base import Plugin class ArtificialIntelligencePlugin(Plugin): type = "artificial-intelligence" - def chat(self, items, **kwargs): - raise NotImplementedError - - def completion(self, items, **kwargs): + def chat_completion(self, items, **kwargs): raise NotImplementedError - def summarization(self, items, **kwargs): + def list_models(self, items, **kwargs): raise NotImplementedError diff --git a/src/dispatch/plugins/dispatch_openai/config.py b/src/dispatch/plugins/dispatch_openai/config.py index 56704e1eda45..c2abe129fd3b 100644 --- a/src/dispatch/plugins/dispatch_openai/config.py +++ b/src/dispatch/plugins/dispatch_openai/config.py @@ -8,7 +8,7 @@ class OpenAIConfiguration(BaseConfigurationModel): api_key: SecretStr = Field(title="API Key", description="Your secret OpenAI API key.") model: str = Field( - "gpt-3.5-turbo", + "gpt-4o", title="Model", description="Available models can be found at https://platform.openai.com/docs/models", ) diff --git a/src/dispatch/plugins/dispatch_openai/plugin.py b/src/dispatch/plugins/dispatch_openai/plugin.py index 81b94990ad0c..4db903c5942f 100644 --- a/src/dispatch/plugins/dispatch_openai/plugin.py +++ b/src/dispatch/plugins/dispatch_openai/plugin.py @@ -33,7 +33,7 @@ class OpenAIPlugin(ArtificialIntelligencePlugin): def __init__(self): self.configuration_schema = OpenAIConfiguration - def completion(self, prompt: str) -> dict: + def chat_completion(self, prompt: str) -> dict: client = OpenAI(api_key=self.api_key) try: