type issues

arjbingly · Mar 30, 2024 · 730a6c4 · 730a6c4
1 parent 17f2385
commit 730a6c4
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"],