alex/LocalAGI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update sample config, fixup sqlite load

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/discord/agent.py
View File

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

12
examples/discord/config.ini.example
View File

@@ -3,7 +3,6 @@ server_id =
api_key = api_key =
[openai] [openai]
organization = fff
api_key = sl-d-d-d api_key = sl-d-d-d
[settings] [settings]
@@ -19,3 +18,14 @@ functions_model = functions
api_base = http://api:8080 api_base = http://api:8080
stablediffusion_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