aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2026-01-19 22:14:28 +01:00
Code Issues Packages Projects Releases Wiki Activity

added support for weaviate embedded

Browse Source
This commit is contained in:
cs0lar
2023-04-12 08:21:41 +01:00
parent 96c5e929be
commit 453b428d33
5 changed files with 25 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/config.py
View File

@@ -66,11 +66,12 @@ class Config(metaclass=Singleton):
self.weaviate_host = os.getenv("WEAVIATE_HOST")
self.weaviate_port = os.getenv("WEAVIATE_PORT")
self.weaviate_protocol = os.getenv("WEAVIATE_PROTOCOL", "http")
self.weaviate_username = os.getenv("WEAVIATE_USERNAME", None)
self.weaviate_password = os.getenv("WEAVIATE_PASSWORD", None)
self.weaviate_scopes = os.getenv("WEAVIATE_SCOPES", None)
self.weaviate_embedded_path = os.getenv('WEAVIATE_EMBEDDED_PATH', '~/.local/share/weaviate')
self.use_weaviate_embedded = os.getenv("USE_WEAVIATE_EMBEDDED", False)
self.use_weaviate_embedded = os.getenv("USE_WEAVIATE_EMBEDDED", "False") == "True"
self.image_provider = os.getenv("IMAGE_PROVIDER")
self.huggingface_api_token = os.getenv("HUGGINGFACE_API_TOKEN")

2
scripts/memory/weaviate.py
View File

@@ -22,7 +22,7 @@ class WeaviateMemory(MemoryProviderSingleton):
def __init__(self, cfg):
auth_credentials = self._build_auth_credentials(cfg)
url = f'{cfg.weaviate_host}:{cfg.weaviate_port}'
url = f'{cfg.weaviate_protocol}://{cfg.weaviate_host}:{cfg.weaviate_port}'
if cfg.use_weaviate_embedded:
self.client = Client(embedded_options=EmbeddedOptions(