From e03ca8854906bdee7d424afd791f7adbcbba37a1 Mon Sep 17 00:00:00 2001
From: Saryev Rustam <r.saryev@indriver.com>
Date: Wed, 12 Jul 2023 03:52:03 +0300
Subject: [PATCH] update project version, add local model images to README, and
 remove default local model from consts.py

---
 README.md               | 6 ++++++
 pyproject.toml          | 2 +-
 talk_codebase/consts.py | 2 --
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index dbbce3b..9c6a2ac 100644
--- a/README.md
+++ b/README.md
@@ -23,11 +23,17 @@ Select model type: Local or OpenAI
 
 <img width="300" alt="select_type" src="https://github.com/rsaryev/talk-codebase/assets/70219513/05196fe5-78ff-44ff-8ca3-0313ccef572a">
 
+OpenAI
 
 If you use the OpenAI model, you need an OpenAI API key. You can get it from [here](https://beta.openai.com/). Then you will be offered a choice of available models.
 
 <img width="300" alt="select" src="https://github.com/rsaryev/talk-codebase/assets/70219513/889ad7c8-a489-4ce8-83af-148b7df09229">
 
+
+Local
+
+<img width="696" alt="Снимок экрана 2023-07-12 в 03 47 58" src="https://github.com/rsaryev/talk-codebase/assets/70219513/16988911-c605-4570-bfb4-4a34a03cd4a1">
+
 If you want some files to be ignored, add them to .gitignore.
 
 ## Reset configuration
diff --git a/pyproject.toml b/pyproject.toml
index 5d494ab..32473bf 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "talk-codebase"
-version = "0.1.43"
+version = "0.1.45"
 description = "talk-codebase is a powerful tool for querying and analyzing codebases."
 authors = ["Saryev Rustam <rustam1997@gmail.com>"]
 readme = "README.md"
diff --git a/talk_codebase/consts.py b/talk_codebase/consts.py
index 44509b6..fcd498b 100644
--- a/talk_codebase/consts.py
+++ b/talk_codebase/consts.py
@@ -14,7 +14,6 @@ MODEL_TYPES = {
     "OPENAI": "openai",
     "LOCAL": "local",
 }
-DEFAULT_LOCAL_MODEL = "orca-mini-3b.ggmlv3.q4_0.bin"
 DEFAULT_MODEL_DIRECTORY = os.path.join(str(Path.home()), ".cache", "gpt4all").replace("\\", "\\\\")
 
 DEFAULT_CONFIG = {
@@ -23,7 +22,6 @@ DEFAULT_CONFIG = {
     "chunk_overlap": "256",
     "k": "1",
     "temperature": "0.7",
-    "local_model_name": DEFAULT_LOCAL_MODEL,
     "model_path": DEFAULT_MODEL_DIRECTORY,
     "n_batch": "8",
 }