This commit is contained in:
William Fu-Hinthorn
2025-02-09 15:34:33 -08:00
parent cc30bc2e06
commit d5d43be824
+6 -2
View File
@@ -43,7 +43,11 @@
"from getpass import getpass\n",
"\n",
"if not os.environ.get(\"ANTHROPIC_API_KEY\"):\n",
" os.environ[\"ANTHROPIC_API_KEY\"] = getpass(\"ANTHROPIC_API_KEY: \")"
" # For the LLM\n",
" os.environ[\"ANTHROPIC_API_KEY\"] = getpass(\"ANTHROPIC_API_KEY: \")\n",
"if not os.environ.get(\"OPENAI_API_KEY\"):\n",
" # For the embedding model\n",
" os.environ[\"OPENAI_API_KEY\"] = getpass(\"OPENAI_API_KEY: \")"
]
},
{
@@ -177,7 +181,7 @@
"from langgraph.store.memory import InMemoryStore\n",
"\n",
"# Memory store\n",
"store = InMemoryStore()"
"store = InMemoryStore(index={\"embed\": \"openai:text-embedding-3-small\"})"
]
},
{