update to gpt4o (#273)

* update to gpt4o * adding prompots dir
Significant-Gravitas · May 17, 2024 · 08d1e51 · 08d1e51
1 parent 8dfebac
commit 08d1e51
Show file tree

Hide file tree

Showing 50 changed files with 4,432 additions and 10 deletions.
diff --git a/codex/common/ai_model.py b/codex/common/ai_model.py
@@ -14,7 +14,7 @@
 def num_tokens_from_messages(messages):
     """Just a rough estimate here."""
     try:
-        encoding = tiktoken.encoding_for_model("gpt-4-turbo")
+        encoding = tiktoken.encoding_for_model("gpt-4o")
     except KeyError:
         print("Warning: model not found. Using cl100k_base encoding.")
         encoding = tiktoken.get_encoding("cl100k_base")

diff --git a/codex/develop/ai_extractor.py b/codex/develop/ai_extractor.py
@@ -13,7 +13,7 @@ class DocumentationExtractor(AIBlock):
     # The name of the prompt template folder in codex/prompts/{model}
     prompt_template_name = "validate/documentation_extractor"
     # Model to use for the LLM
-    model = "gpt-4-turbo"
+    model = "gpt-4o"
     # Should we force the LLM to reply in JSON
     is_json_response = False
 

diff --git a/codex/develop/develop.py b/codex/develop/develop.py
@@ -72,7 +72,7 @@ def parse_requirements(requirements_str: str) -> List[Package]:
 class DevelopAIBlock(AIBlock):
     developement_phase: DevelopmentPhase = DevelopmentPhase.DEVELOPMENT
     prompt_template_name = "develop"
-    model = "gpt-4-turbo"
+    model = "gpt-4o"
     language = "python"
 
     async def validate(

diff --git a/codex/interview/ai_interview.py b/codex/interview/ai_interview.py
@@ -19,7 +19,7 @@ class InterviewBlock(AIBlock):
     """
 
     prompt_template_name = "interview/understand"
-    model = "gpt-4-turbo"
+    model = "gpt-4o"
     is_json_response = True
     pydantic_object = UndestandRequest
 

diff --git a/codex/interview/ai_interview_update.py b/codex/interview/ai_interview_update.py
@@ -30,7 +30,7 @@ class InterviewUpdateBlock(AIBlock):
     """
 
     prompt_template_name = "interview/update"
-    model = "gpt-4-turbo"
+    model = "gpt-4o"
     is_json_response = True
     pydantic_object = UpdateUnderstanding
 

diff --git a/codex/interview/ai_module.py b/codex/interview/ai_module.py
@@ -18,7 +18,7 @@ class ModuleGenerationBlock(codex.common.ai_block.AIBlock):
     # The name of the prompt template folder in codex/prompts/{model}
     prompt_template_name = "interview/module"
     # Model to use for the LLM
-    model = "gpt-4-turbo"
+    model = "gpt-4o"
     # Should we force the LLM to reply in JSON
     is_json_response = True
     # If we are using is_json_response, what is the response model