ignore generated proto files

ricardo-larosa · Aug 15, 2024 · 6137948 · 6137948
1 parent 5e280cd
commit 6137948
Show file tree

Hide file tree

Showing 11 changed files with 12 additions and 12 deletions.
diff --git a/.gitignore b/.gitignore
@@ -158,6 +158,6 @@ cython_debug/
 .ragatouille
 
 # Generated proto files
-# colbert_rag/proto/*_pb2.py
-# colbert_rag/proto/*_pb2_grpc.py
-# colbert_rag/proto/*_pb2.pyi
+/colbert_rag/proto/colbertrag_pb2_grpc.py
+/colbert_rag/proto/colbertrag_pb2.py
+/colbert_rag/proto/colbertrag_pb2.pyi
diff --git a/colbert_rag/__init__.py b/colbert_rag/__init__.py
@@ -10,4 +10,4 @@
            'FastAPIServer', 
            'Request', 
            'Response', 
-           'Document']
+           'Document']
diff --git a/colbert_rag/config.py b/colbert_rag/config.py
@@ -5,4 +5,4 @@
 COLBERTRAG_MAX_WORKERS = 10
 COLBERTRAG_GRPC_PORT = 50051
 COLBERTRAG_FASTAPI_PORT = 8000
-COLBERTRAG_HOST = '0.0.0.0'
+COLBERTRAG_HOST = '0.0.0.0'
diff --git a/colbert_rag/data/git_repo.py b/colbert_rag/data/git_repo.py
@@ -65,3 +65,4 @@ def get_collections(
                     collections[language][i].append(item)
 
         return dict(collections)
+
diff --git a/colbert_rag/indexer/base.py b/colbert_rag/indexer/base.py
@@ -44,4 +44,4 @@ def update_index(
         new_document_metadatas=new_document_metadatas,
         document_splitter_fn=document_splitter_fn,
         split_documents=split_documents,
-        use_faiss=use_faiss)
+        use_faiss=use_faiss)
diff --git a/colbert_rag/indexer/git_repo.py b/colbert_rag/indexer/git_repo.py
@@ -69,4 +69,4 @@ def index_git_repo(
             use_faiss=use_faiss
         )
 
-    return path
+    return path
diff --git a/colbert_rag/models.py b/colbert_rag/models.py
@@ -10,4 +10,4 @@ class Document(BaseModel):
     metadata: Dict[str, str]
 
 class Response(BaseModel):
-    documents: List[Document]
+    documents: List[Document]
diff --git a/colbert_rag/proto/__init__.py b/colbert_rag/proto/__init__.py
@@ -1,3 +1,3 @@
 from .wrapper import colbertrag_pb2, colbertrag_pb2_grpc
 
-__all__ = ['colbertrag_pb2', 'colbertrag_pb2_grpc']
+__all__ = ['colbertrag_pb2', 'colbertrag_pb2_grpc']
diff --git a/colbert_rag/proto/wrapper.py b/colbert_rag/proto/wrapper.py
@@ -9,4 +9,4 @@
 from . import colbertrag_pb2_grpc
 
 # Re-export the modules
-__all__ = ['colbertrag_pb2', 'colbertrag_pb2_grpc']
+__all__ = ['colbertrag_pb2', 'colbertrag_pb2_grpc']
diff --git a/mypy.ini b/mypy.ini
@@ -13,4 +13,4 @@ exclude = ^colbert_rag/proto
 disallow_untyped_defs = True
 
 [mypy-tests.*]
-disallow_untyped_defs = False
+disallow_untyped_defs = False
diff --git a/proto/colbertrag.proto b/proto/colbertrag.proto
@@ -19,4 +19,3 @@ message Document {
 message Response {
   repeated Document documents = 1;
 }
-
Original file line number	Diff line number	Diff line change
Expand Up		@@ -65,3 +65,4 @@ def get_collections(
		collections[language][i].append(item)

		return dict(collections)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -19,4 +19,3 @@ message Document {
		message Response {
		repeated Document documents = 1;
		}