Skip to content

Commit 7e9bf45

Browse files
authored
Update src_core.py
1 parent 302efe3 commit 7e9bf45

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src_core.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,9 +87,9 @@
8787
go_codes_out = extract_relevant_part.get_relevant_part(go_code_in, parser)
8888

8989
# Initialize tokenizer and models
90-
tokenizer = Anthropic().tokenizer
91-
llm = Anthropic(model="claude-3-haiku-20240307")
92-
#llm = Gemini(model="models/gemini-pro")
90+
#tokenizer = Anthropic().tokenizer
91+
#llm = Anthropic(model="claude-3-haiku-20240307")
92+
llm = Gemini(model="models/gemini-pro")
9393
embed_model = GeminiEmbedding(model_name="models/text-embedding-004")
9494

9595
# Initialize vector store and index

0 commit comments

Comments
 (0)