OnurKerimoglu commited on
Commit
3446837
·
1 Parent(s): ab24c7b

nb/chat: updated openai modelname and cleaned up key retreival

Browse files
Files changed (1) hide show
  1. 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
- "_ = load_dotenv(find_dotenv()) # read local .env file\n",
28
- "# openai.api_key = os.environ['OPENAI_API_KEY']\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,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-3.5-turbo\", temperature=0)"
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
  {