Updates AI plugin

src/dispatch/plugins/bases/artificial_intelligence.py

@@ -5,17 +5,15 @@
     :license: Apache, see LICENSE for more details.
 .. moduleauthor:: Marc Vilanova <mvilanova@netflix.com>
 """
+
 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

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",
     )

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: