Spaces:
Running
Running
Commit
·
3446837
1
Parent(s):
ab24c7b
nb/chat: updated openai modelname and cleaned up key retreival
Browse files- notebooks/chat.ipynb +5 -5
notebooks/chat.ipynb
CHANGED
|
@@ -24,16 +24,16 @@
|
|
| 24 |
"import sys\n",
|
| 25 |
"from dotenv import load_dotenv, find_dotenv\n",
|
| 26 |
"\n",
|
| 27 |
-
"
|
| 28 |
-
"
|
| 29 |
"\n",
|
| 30 |
"abscurdir = os.path.abspath(os.curdir)\n",
|
| 31 |
"docsdir = os.path.join(os.path.dirname(abscurdir), 'docs')\n",
|
| 32 |
"existing_dbname = 'chroma_20241124_132314'\n",
|
| 33 |
"persist_directory = os.path.join(docsdir, existing_dbname)\n",
|
| 34 |
"collection_name = 'MLbooks'\n",
|
| 35 |
-
"llm_name = 'ollama3.2.1b'\n",
|
| 36 |
-
"
|
| 37 |
]
|
| 38 |
},
|
| 39 |
{
|
|
@@ -84,7 +84,7 @@
|
|
| 84 |
" llm = Ollama(model=\"llama3.2:1b\", temperature=0)\n",
|
| 85 |
"elif llm_name == 'openai':\n",
|
| 86 |
" from langchain_openai import ChatOpenAI\n",
|
| 87 |
-
" llm = ChatOpenAI(model_name=\"gpt-
|
| 88 |
]
|
| 89 |
},
|
| 90 |
{
|
|
|
|
| 24 |
"import sys\n",
|
| 25 |
"from dotenv import load_dotenv, find_dotenv\n",
|
| 26 |
"\n",
|
| 27 |
+
"# read local .env file (that should contain an \"OPENAI_API_KEY\")\n",
|
| 28 |
+
"_ = load_dotenv(find_dotenv()) \n",
|
| 29 |
"\n",
|
| 30 |
"abscurdir = os.path.abspath(os.curdir)\n",
|
| 31 |
"docsdir = os.path.join(os.path.dirname(abscurdir), 'docs')\n",
|
| 32 |
"existing_dbname = 'chroma_20241124_132314'\n",
|
| 33 |
"persist_directory = os.path.join(docsdir, existing_dbname)\n",
|
| 34 |
"collection_name = 'MLbooks'\n",
|
| 35 |
+
"# llm_name = 'ollama3.2.1b'\n",
|
| 36 |
+
"llm_name = 'openai'"
|
| 37 |
]
|
| 38 |
},
|
| 39 |
{
|
|
|
|
| 84 |
" llm = Ollama(model=\"llama3.2:1b\", temperature=0)\n",
|
| 85 |
"elif llm_name == 'openai':\n",
|
| 86 |
" from langchain_openai import ChatOpenAI\n",
|
| 87 |
+
" llm = ChatOpenAI(model_name=\"gpt-4o-mini\", temperature=0)"
|
| 88 |
]
|
| 89 |
},
|
| 90 |
{
|