New GPT4 Turbo deployment (#2)

gnosis · Apr 15, 2024 · 30e2306 · 30e2306
1 parent 4d74c0f
commit 30e2306
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/predictionprophet_deployment/agents/prophet_agent/deploy.py b/predictionprophet_deployment/agents/prophet_agent/deploy.py
@@ -135,11 +135,17 @@ class DeployablePredictionProphetGPT3Agent(DeployableAgentER):
     agent = PredictionProphetAgent(model="gpt-3.5-turbo-0125")
 
 
-class DeployablePredictionProphetGPT4Agent(DeployableAgentER):
+class DeployablePredictionProphetGPT4TurboPreviewAgent(DeployableAgentER):
     agent = PredictionProphetAgent(model="gpt-4-0125-preview")
     # Limit to just 1, because so far it seems that 20x higher costs aren't justified by the prediction performance.
     max_markets_per_run = 1
 
 
+class DeployablePredictionProphetGPT4TurboFinalAgent(DeployableAgentER):
+    agent = PredictionProphetAgent(model="gpt-4-turbo-2024-04-09")
+    # Limit to just 1, because so far it seems that 20x higher costs aren't justified by the prediction performance.
+    max_markets_per_run = 1
+
+
 class DeployableOlasEmbeddingOAAgent(DeployableAgentER):
     agent = OlasAgent(model="gpt-3.5-turbo-0125", embedding_model=EmbeddingModel.openai)
diff --git a/predictionprophet_deployment/run_agent.py b/predictionprophet_deployment/run_agent.py
@@ -14,19 +14,22 @@
 from predictionprophet_deployment.agents.prophet_agent.deploy import (
     DeployableOlasEmbeddingOAAgent,
     DeployablePredictionProphetGPT3Agent,
-    DeployablePredictionProphetGPT4Agent,
+    DeployablePredictionProphetGPT4TurboFinalAgent,
+    DeployablePredictionProphetGPT4TurboPreviewAgent,
 )
 
 
 class RunnableAgent(str, Enum):
     prophet_gpt3 = "prophet_gpt3"
     prophet_gpt4 = "prophet_gpt4"
+    prophet_gpt4_final = "prophet_gpt4_final"
     olas_embedding_oa = "olas_embedding_oa"
 
 
 RUNNABLE_AGENTS = {
     RunnableAgent.prophet_gpt3: DeployablePredictionProphetGPT3Agent,
-    RunnableAgent.prophet_gpt4: DeployablePredictionProphetGPT4Agent,
+    RunnableAgent.prophet_gpt4: DeployablePredictionProphetGPT4TurboPreviewAgent,
+    RunnableAgent.prophet_gpt4_final: DeployablePredictionProphetGPT4TurboFinalAgent,
     RunnableAgent.olas_embedding_oa: DeployableOlasEmbeddingOAAgent,
 }