Skip to content

Commit e598c55

Browse files
committed
fixed model version to latest
1 parent 1473c69 commit e598c55

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: vulnhuntr/__main__.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -284,11 +284,11 @@ def extract_between_tags(tag: str, string: str, strip: bool = False) -> list[str
284284
def initialize_llm(llm_arg: str, system_prompt: str = "") -> Claude | ChatGPT | Ollama:
285285
llm_arg = llm_arg.lower()
286286
if llm_arg == 'claude':
287-
anth_model = os.getenv("ANTHROPIC_MODEL", "claude-3-5-sonnet-20240620")
287+
anth_model = os.getenv("ANTHROPIC_MODEL", "claude-3-5-sonnet-latest")
288288
anth_base_url = os.getenv("ANTHROPIC_BASE_URL", "https://api.anthropic.com")
289289
llm = Claude(anth_model, anth_base_url, system_prompt)
290290
elif llm_arg == 'gpt':
291-
openai_model = os.getenv("OPENAI_MODEL", "gpt-4o-2024-08-06")
291+
openai_model = os.getenv("OPENAI_MODEL", "chatgpt-4o-latest")
292292
openai_base_url = os.getenv("OPENAI_BASE_URL", "https://api.openai.com/v1")
293293
llm = ChatGPT(openai_model, openai_base_url, system_prompt)
294294
elif llm_arg == 'ollama':

0 commit comments

Comments
 (0)