:question_mark: dunno why this wants to be commited

br-data · Dec 30, 2023 · d38db67 · d38db67
1 parent b531c63
commit d38db67
Show file tree

Hide file tree

Showing 18 changed files with 3 additions and 4 deletions.
diff --git a/.github/workflows/documentation.yml b/.github/workflows/documentation.yml
diff --git a/.readthedocs.yaml b/.readthedocs.yaml
diff --git a/LICENSE b/LICENSE
diff --git a/README.md b/README.md
diff --git a/dist/brdata-rag-tools-0.0.1.tar.gz b/dist/brdata-rag-tools-0.0.1.tar.gz
diff --git a/dist/brdata_rag_tools-0.0.1-py3-none-any.whl b/dist/brdata_rag_tools-0.0.1-py3-none-any.whl
diff --git a/docs/Makefile b/docs/Makefile
diff --git a/docs/make.bat b/docs/make.bat
diff --git a/docs/requirements.txt b/docs/requirements.txt
diff --git a/docs/source/conf.py b/docs/source/conf.py
diff --git a/docs/source/index.rst b/docs/source/index.rst
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,10 +4,11 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "brdata-rag-tools"
-version = "0.0.3"
+version = "0.0.5"
 authors = [
     { name = "Marco Lehner", email = "[email protected]" },
 ]
+
 description = "Improve development of retrieval augmented generation (RAG) applications at the BR AI + Automation Lab."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -80,7 +81,6 @@ dependencies = [
     "sphinxcontrib-qthelp==1.0.6",
     "sphinxcontrib-serializinghtml==1.1.9",
     "SQLAlchemy==2.0.23",
-    "sqlite-vss==0.1.2",
     "tiktoken==0.5.2",
     "tomli==2.0.1",
     "tqdm==4.66.1",

diff --git a/src/__init__.py b/src/__init__.py
diff --git a/src/brdata_rag_tools/__init__.py b/src/brdata_rag_tools/__init__.py
diff --git a/src/brdata_rag_tools/datastructures.py b/src/brdata_rag_tools/datastructures.py
diff --git a/src/brdata_rag_tools/main.py b/src/brdata_rag_tools/main.py
@@ -7,7 +7,6 @@
 from models import LLM
 
 if __name__ == "__main__":
-
     # load model
     language_model = LLM(model_name=LLMName.GPT35TURBO)
     embed_type = EmbeddingType.SENTENCE_TRANSFORMERS
@@ -43,7 +42,7 @@ class Podcast(EmbeddingTable):
     database.write_rows([podcast1, podcast2, podcast3], create_embeddings = True)
 
     # create the prompt and template
-    user_prompt = "Gib mir spannende Podcasts zum Thema Musik."
+    user_prompt = "Welche spannenden Podcasts zum Thema Musik kannst du mir empfehlen?"
 
     prompt_template = ("Du bist der Podcast-Experte des Bayerischen Rundfunks. "
                        "Eine Userin stellt dir folgende Frage:\n"

diff --git a/src/brdata_rag_tools/models.py b/src/brdata_rag_tools/models.py
diff --git a/test/test_databases.py b/test/test_databases.py