Solve pylint complains

Maclenn77 · Dec 9, 2023 · c3e9f44 · c3e9f44
1 parent e303e75
commit c3e9f44
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/chroma_client.py b/src/chroma_client.py
@@ -13,6 +13,14 @@ def __init__(self, api_key, path="tmp/chroma"):
         self.api_key = api_key
         self.client.heartbeat()
 
+    def get_collection(self, name):
+        """Get a Chroma collection."""
+        try:
+            collection = self.client.get_collection(name=name)
+            return collection
+        except AttributeError:
+            return st.error("An error ocurred while getting the collection.")
+
     def create_collection(self, name):
         """Create a Chroma collection."""
         try: