pytrilogy 0.0.3.119__tar.gz → 0.0.3.120__tar.gz

{pytrilogy-0.0.3.119/pytrilogy.egg-info → pytrilogy-0.0.3.120}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytrilogy
-Version: 0.0.3.119
+Version: 0.0.3.120
 Summary: Declarative, typed query language that compiles to SQL.
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3

{pytrilogy-0.0.3.119 → pytrilogy-0.0.3.120/pytrilogy.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytrilogy
-Version: 0.0.3.119
+Version: 0.0.3.120
 Summary: Declarative, typed query language that compiles to SQL.
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3

{pytrilogy-0.0.3.119 → pytrilogy-0.0.3.120}/trilogy/__init__.py

@@ -4,6 +4,6 @@ from trilogy.dialect.enums import Dialects
 from trilogy.executor import Executor
 from trilogy.parser import parse
 
-__version__ = "0.0.3.119"
+__version__ = "0.0.3.120"
 
 __all__ = ["parse", "Executor", "Dialects", "Environment", "CONFIG"]

{pytrilogy-0.0.3.119 → pytrilogy-0.0.3.120}/trilogy/ai/conversation.py

@@ -61,7 +61,7 @@ class Conversation:
         if "```" in content:
             parts = content.split("```")
             if len(parts) >= 3:
-                return parts[1].strip()
+                return parts[-2].strip()
         return content
 
     def generate_query(
@@ -78,7 +78,7 @@ class Conversation:
             if not response.strip()[-1] == ";":
                 response += ";"
             try:
-                env, raw = environment.parse(response)
+                _, raw = environment.parse(response)
                 process_query(statement=raw[-1], environment=environment)
                 return response
             except (
@@ -90,10 +90,9 @@ class Conversation:
             ) as e2:
                 e = e2
                 self.add_message(
-                    f"The previous response could not be parsed due to the error: {str(e)}. Please generate a new query with the issues fixed. Use the same response format.",
+                    f"Your extracted response - {response} - could not be parsed due to the error: {str(e)}. Please generate a new query with the issues fixed. Use the same response format.",
                     role="user",
                 )
-
         raise Exception(
             f"Failed to generate a valid query after {attempts} attempts. Last error: {str(e)}. Full conversation: {self.messages}"
         )

{pytrilogy-0.0.3.119 → pytrilogy-0.0.3.120}/trilogy/core/models/environment.py

@@ -215,7 +215,7 @@ class Environment(BaseModel):
     functions: Dict[str, CustomFunctionFactory] = Field(default_factory=dict)
     data_types: Dict[str, CustomType] = Field(default_factory=dict)
     named_statements: Dict[str, SelectLineage] = Field(default_factory=dict)
-    imports: Dict[str, list[Import]] = Field(
+    imports: defaultdict[str, list[Import]] = Field(
         default_factory=lambda: defaultdict(list)  # type: ignore
     )
     namespace: str = DEFAULT_NAMESPACE
@@ -252,7 +252,7 @@ class Environment(BaseModel):
             concepts=self.concepts.duplicate(),
             functions=dict(self.functions),
             data_types=dict(self.data_types),
-            imports=dict(self.imports),
+            imports=defaultdict(list, self.imports),
             namespace=self.namespace,
             working_path=self.working_path,
             environment_config=self.config.model_copy(deep=True),
@@ -262,6 +262,7 @@ class Environment(BaseModel):
             alias_origin_lookup={
                 k: v.duplicate() for k, v in self.alias_origin_lookup.items()
             },
+            env_file_path=self.env_file_path,
         )
 
     def _add_path_concepts(self):

{pytrilogy-0.0.3.119 → pytrilogy-0.0.3.120}/trilogy/core/processing/discovery_utility.py

@@ -40,15 +40,15 @@ def calculate_effective_parent_grain(
             pairs = join.concept_pairs or []
             for key in pairs:
                 left = key.existing_datasource
-                logger.info(f"adding left grain {left.grain} for join key {key.left}")
+                logger.debug(f"adding left grain {left.grain} for join key {key.left}")
                 grain += left.grain
                 seen.add(left.name)
             keys = [key.right for key in pairs]
             join_grain = BuildGrain.from_concepts(keys)
             if join_grain == join.right_datasource.grain:
-                logger.info(f"irrelevant right join {join}, does not change grain")
+                logger.debug(f"irrelevant right join {join}, does not change grain")
             else:
-                logger.info(
+                logger.debug(
                     f"join changes grain, adding {join.right_datasource.grain} to {grain}"
                 )
                 grain += join.right_datasource.grain
@@ -68,7 +68,7 @@ def calculate_effective_parent_grain(
                     ]
                 )
             ):
-                logger.info(f"adding unjoined grain {x.grain} for datasource {x.name}")
+                logger.debug(f"adding unjoined grain {x.grain} for datasource {x.name}")
                 grain += x.grain
         return grain
     else:

{pytrilogy-0.0.3.119 → pytrilogy-0.0.3.120}/trilogy/core/processing/nodes/base_node.py

@@ -258,10 +258,14 @@ class StrategyNode:
         self.partial_lcl = LooseBuildConceptList(concepts=partials)
         return partials
 
-    def add_output_concepts(self, concepts: List[BuildConcept], rebuild: bool = True):
+    def add_output_concepts(
+        self, concepts: List[BuildConcept], rebuild: bool = True, unhide: bool = True
+    ):
         for concept in concepts:
             if concept.address not in self.output_lcl.addresses:
                 self.output_concepts.append(concept)
+            if unhide and concept.address in self.hidden_concepts:
+                self.hidden_concepts.remove(concept.address)
         self.output_lcl = LooseBuildConceptList(concepts=self.output_concepts)
         if rebuild:
             self.rebuild_cache()