fix collection ref

Josh-XT · Nov 19, 2024 · 3085cf8 · 3085cf8
1 parent fec5b61
commit 3085cf8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/agixt/XT.py b/agixt/XT.py
@@ -95,11 +95,11 @@ def __init__(
         self.failures = 0
         self.input_tokens = 0
         self.file_reader = None
-        if self.conversation_id:
+        if self.collection_id is not None:
             self.file_reader = FileReader(
                 agent_name=self.agent_name,
                 agent_config=self.agent.AGENT_CONFIG,
-                collection_number=self.conversation_id,
+                collection_number=self.collection_id,
                 ApiClient=self.ApiClient,
                 user=self.user_email,
             )
@@ -715,7 +715,7 @@ async def learn_from_file(
         file_url: str = "",
         file_name: str = "",
         user_input: str = "",
-        collection_id: str = "1",
+        collection_id: str = "0",
     ):
         """
         Learn from a file