File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 22from pydantic import BaseModel
33from enum import Enum
44
5+
56class ModelProvider (Enum ):
67 ANTHROPIC = "anthropic"
78 OPENAI = "openai"
89 TOGETHER = "together"
910 GOOGLE = "google"
1011
12+
1113class ModelTemplate (BaseModel ):
1214 name : str
1315 max_concurrency : int
1416 requires_chain_of_thought : bool
1517 uses_system_prompt : bool
1618 provider : ModelProvider
1719
20+
1821ALL_MODELS = [
1922 ModelTemplate (
2023 name = "claude-3-5-sonnet-latest" ,
@@ -71,13 +74,14 @@ class ModelTemplate(BaseModel):
7174 requires_chain_of_thought = True ,
7275 uses_system_prompt = False ,
7376 provider = ModelProvider .GOOGLE ,
74- )
77+ ),
7578]
7679MODELS_BY_NAME = {model .name : model for model in ALL_MODELS }
7780
81+
7882class ConvexCodegenModel :
7983 def generate (self , user_prompt : str ) -> dict [str , str ]:
8084 raise NotImplementedError ()
8185
8286
83- SYSTEM_PROMPT = "You are convexbot, a highly advanced AI programmer specialized in creating backend systems using Convex."
87+ SYSTEM_PROMPT = "You are convexbot, a highly advanced software engineer specialized in creating applications using Convex and TypeScript ."
You can’t perform that action at this time.
0 commit comments