diff --git a/predictionprophet_deployment/agents/prophet_agent/deploy.py b/predictionprophet_deployment/agents/prophet_agent/deploy.py index 41a19ec..b575760 100644 --- 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 index 59e61ed..57c91d1 100644 --- 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, }