Bläddra i källkod

Fix/dont print the entire text when data type is text (#605)

David Talson 1 år sedan
förälder
incheckning
5e06a0d001
1 ändrade filer med 8 tillägg och 2 borttagningar
  1. 8 2
      embedchain/embedchain.py

+ 8 - 2
embedchain/embedchain.py

@@ -284,7 +284,10 @@ class EmbedChain(JSONSerializable):
             data_dict = {id: value for id, value in data_dict.items() if id not in existing_ids}
 
             if not data_dict:
-                print(f"All data from {src} already exists in the database.")
+                src_copy = src
+                if len(src_copy) > 50:
+                    src_copy = src[:50] + "..."
+                print(f"All data from {src_copy} already exists in the database.")
                 # Make sure to return a matching return type
                 return [], [], [], 0
 
@@ -418,7 +421,10 @@ class EmbedChain(JSONSerializable):
             data_dict = {id: value for id, value in data_dict.items() if id not in existing_ids}
 
             if not data_dict:
-                print(f"All data from {src} already exists in the database.")
+                src_copy = src
+                if len(src_copy) > 50:
+                    src_copy = src[:50] + "..."
+                print(f"All data from {src_copy} already exists in the database.")
                 # Make sure to return a matching return type
                 return [], [], [], 0