Skip to content
This repository has been archived by the owner on Aug 6, 2024. It is now read-only.

New GPT4 Turbo deployment #2

Merged
merged 1 commit into from
Apr 15, 2024
Merged

New GPT4 Turbo deployment #2

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion predictionprophet_deployment/agents/prophet_agent/deploy.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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)
7 changes: 5 additions & 2 deletions predictionprophet_deployment/run_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

"final" because next will be gpt5? 😁

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,
}

Expand Down
Loading