Update sample config, fixup sqlite load

This commit is contained in:
mudler
2023-08-26 00:48:40 +02:00
parent 414f9ca765
commit 87736e464a
2 changed files with 17 additions and 8 deletions

View File

@@ -46,13 +46,12 @@ MEMORY_CHUNK_OVERLAP = int(config["agent"]["memory_chunk_overlap"])
MEMORY_RESULTS = int(config["agent"]["memory_results"])
MEMORY_SEARCH_TYPE = config["agent"]["memory_search_type"]
if not os.environ.get("PYSQL_HACK", "false") == "false":
# these three lines swap the stdlib sqlite3 lib with the pysqlite3 package for chroma
__import__('pysqlite3')
import sys
sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
if MILVUS_HOST == "":
if not os.environ.get("PYSQL_HACK", "false") == "false":
# these three lines swap the stdlib sqlite3 lib with the pysqlite3 package for chroma
__import__('pysqlite3')
import sys
sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
from langchain.vectorstores import Chroma
else:
from langchain.vectorstores import Milvus

View File

@@ -3,7 +3,6 @@ server_id =
api_key =
[openai]
organization = fff
api_key = sl-d-d-d
[settings]
@@ -18,4 +17,15 @@ tts_api_base = http://api:8080
functions_model = functions
api_base = http://api:8080
stablediffusion_api_base = http://api:8080
stablediffusion_model = stablediffusion
stablediffusion_model = stablediffusion
embeddings_model = all-MiniLM-L6-v2
embeddings_api_base = http://api:30316/v1
persistent_dir = /tmp/data
db_dir = /tmp/data/db
milvus_host =
milvus_port =
memory_collection = localai
memory_chunk_size = 600
memory_chunk_overlap = 110
memory_results = 3
memory_search_type = mmr