palimpzest 0.7.10__tar.gz → 0.7.11__tar.gz

{palimpzest-0.7.10/src/palimpzest.egg-info → palimpzest-0.7.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: palimpzest
-Version: 0.7.10
+Version: 0.7.11
 Summary: Palimpzest is a system which enables anyone to process AI-powered analytical queries simply by defining them in a declarative language
 Author-email: MIT DSG Semantic Management Lab <michjc@csail.mit.edu>
 Project-URL: homepage, https://palimpzest.org

{palimpzest-0.7.10 → palimpzest-0.7.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "palimpzest"
-version = "0.7.10"
+version = "0.7.11"
 description = "Palimpzest is a system which enables anyone to process AI-powered analytical queries simply by defining them in a declarative language"
 readme = "README.md"
 requires-python = ">=3.8"

{palimpzest-0.7.10 → palimpzest-0.7.11}/src/palimpzest/query/generators/generators.py

@@ -215,14 +215,15 @@ class BaseGenerator(Generic[ContextType, InputType], ABC):
 
             # TODO: wrap non-list outputs in a list if expected output is a list
 
-            # common error: if the output is a singleton list which contains a list, but the expected field type
+            # common error for one-to-one: if the output is a singleton list which contains a list, but the expected field type
             # is a list of strings, or a list of floats, i.e. not a list of lists; then extract the inner list
-            for field, field_type in fields.items():
-                answer = field_answers[field]
-                field_type_is_not_list_of_lists = isinstance(field_type, ListField) and not issubclass(field_type.element_type, ListField)
-                answer_is_list_of_lists = isinstance(answer, list) and len(answer) == 1 and isinstance(answer[0], list)
-                if field_type_is_not_list_of_lists and answer_is_list_of_lists:
-                    field_answers[field] = answer[0]
+            if self.cardinality == Cardinality.ONE_TO_ONE:
+                for field, field_type in fields.items():
+                    answer = field_answers[field]
+                    field_type_is_not_list_of_lists = isinstance(field_type, ListField) and not issubclass(field_type.element_type, ListField)
+                    answer_is_list_of_lists = isinstance(answer, list) and len(answer) == 1 and isinstance(answer[0], list)
+                    if field_type_is_not_list_of_lists and answer_is_list_of_lists:
+                        field_answers[field] = answer[0]
 
             # prepare the field answers to match the expected output and return
             return self._prepare_field_answers(field_answers, fields)
@@ -416,6 +417,8 @@ class BaseGenerator(Generic[ContextType, InputType], ABC):
                 prompt += message["content"] + "\n" if message["type"] == "text" else "<image>\n"
         logger.debug(f"PROMPT:\n{prompt}")
         logger.debug(Fore.GREEN + f"{completion_text}\n" + Style.RESET_ALL)
+        print(f"PROMPT:\n{prompt}")
+        print(Fore.GREEN + f"{completion_text}\n" + Style.RESET_ALL)
 
         # parse reasoning
         reasoning = None

{palimpzest-0.7.10 → palimpzest-0.7.11/src/palimpzest.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: palimpzest
-Version: 0.7.10
+Version: 0.7.11
 Summary: Palimpzest is a system which enables anyone to process AI-powered analytical queries simply by defining them in a declarative language
 Author-email: MIT DSG Semantic Management Lab <michjc@csail.mit.edu>
 Project-URL: homepage, https://palimpzest.org