Update sample config, fixup sqlite load
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
|
||||
Reference in New Issue
Block a user