Merge pull request #63 from arjbingly/tests

type issues
arjbingly · Mar 30, 2024 · 03a421b · 03a421b
2 parents b51df79 + 730a6c4
commit 03a421b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/grag/components/vectordb/chroma_client.py b/src/grag/components/vectordb/chroma_client.py
@@ -44,8 +44,8 @@ class ChromaClient(VectorDB):
 
     def __init__(
         self,
-        host: str = chroma_conf["host"],
-        port: str = chroma_conf["port"],
+        host: int = chroma_conf["host"],
+        port: int = chroma_conf["port"],
         collection_name: str = chroma_conf["collection_name"],
         embedding_type: str = chroma_conf["embedding_type"],
         embedding_model: str = chroma_conf["embedding_model"],