sie-haystack 0.6.18__tar.gz → 0.6.20__tar.gz

{sie_haystack-0.6.18 → sie_haystack-0.6.20}/.gitignore

@@ -244,6 +244,10 @@ override.tf.json
 # tfvars files may contain secrets
 *.tfvars
 *.tfvars.json
+# Reviewed worker-AMI release contracts contain only immutable public package,
+# AMI, and Terraform backend coordinates; the release workflow accepts only
+# committed files from this exact internal path.
+!deploy/terraform/aws/internal-examples/cloud-benchmark-runner/worker-ami/releases/*.tfvars.json
 # Exception: this edge tfvars carries only a Secrets Manager ARN (a resource
 # identifier, not a secret) so the staging-us edge apply needs no manual -var.
 !deploy/cloud/terraform/aws-edge/staging-us.tfvars
@@ -272,4 +276,7 @@ tmp/
 .local/
 
 .requirements-modal.txt
+b6_results.json
 packages/sie_cloud/deploy/manifests/archive/
+# Raw dev-cloud load-test reports are local artifacts, not source-controlled baselines.
+packages/sie_cloud/bench/results/dev-us-*/

{sie_haystack-0.6.18 → sie_haystack-0.6.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sie-haystack
-Version: 0.6.18
+Version: 0.6.20
 Summary: SIE integration for Haystack
 Author-email: Superlinked <dev@superlinked.com>
 License: Apache-2.0
@@ -75,3 +75,24 @@ ranker = SIERanker(
 result = ranker.run(query="What is Python?", documents=embedded_docs, top_k=3)
 ranked_docs = result["documents"]
 ```
+
+Relation models such as GLiREL classify relations between supplied entity
+spans. Extract those spans first, then pass them through the relation
+extractor's `entities` input:
+
+```python
+from haystack_integrations.components.extractors.sie import SIEExtractor
+
+text = "Tim Cook is the CEO of Apple Inc."
+entity_extractor = SIEExtractor(
+    model="urchade/gliner_multi-v2.1",
+    labels=["person", "organization"],
+)
+relation_extractor = SIEExtractor(
+    model="jackboyla/glirel-large-v0",
+    labels=["ceo_of"],
+)
+
+entities = entity_extractor.run(text=text)["entities"]
+result = relation_extractor.run(text=text, entities=entities)
+```

{sie_haystack-0.6.18 → sie_haystack-0.6.20}/README.md

@@ -53,3 +53,24 @@ ranker = SIERanker(
 result = ranker.run(query="What is Python?", documents=embedded_docs, top_k=3)
 ranked_docs = result["documents"]
 ```
+
+Relation models such as GLiREL classify relations between supplied entity
+spans. Extract those spans first, then pass them through the relation
+extractor's `entities` input:
+
+```python
+from haystack_integrations.components.extractors.sie import SIEExtractor
+
+text = "Tim Cook is the CEO of Apple Inc."
+entity_extractor = SIEExtractor(
+    model="urchade/gliner_multi-v2.1",
+    labels=["person", "organization"],
+)
+relation_extractor = SIEExtractor(
+    model="jackboyla/glirel-large-v0",
+    labels=["ceo_of"],
+)
+
+entities = entity_extractor.run(text=text)["entities"]
+result = relation_extractor.run(text=text, entities=entities)
+```

{sie_haystack-0.6.18 → sie_haystack-0.6.20}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sie-haystack"
-version = "0.6.18"
+version = "0.6.20"
 description = "SIE integration for Haystack"
 readme = "README.md"
 requires-python = ">=3.10"

{sie_haystack-0.6.18 → sie_haystack-0.6.20}/src/sie_haystack/extractors.py

@@ -10,6 +10,7 @@ from dataclasses import dataclass
 from typing import Any
 
 from haystack import component
+from sie_sdk.types import Item
 
 
 @dataclass
@@ -125,23 +126,40 @@ class SIEExtractor:
         self,
         text: str,
         labels: list[str] | None = None,
+        entities: list[Entity | dict[str, Any]] | None = None,
     ) -> dict[str, list]:
         """Extract structured information from text.
 
         Args:
             text: The text to extract from.
             labels: Override the configured labels for this call.
+            entities: Entity spans for relation extraction models such as GLiREL.
+                Each span uses character offsets with an exclusive ``end``.
 
         Returns:
             Dictionary with entities, relations, classifications, and objects.
         """
-        from sie_sdk.types import Item
-
         effective_labels = labels if labels is not None else self._labels
+        metadata = None
+        if entities is not None:
+            metadata = {
+                "entities": [
+                    dict(entity)
+                    if isinstance(entity, dict)
+                    else {
+                        "text": entity.text,
+                        "label": entity.label,
+                        "score": entity.score,
+                        "start": entity.start,
+                        "end": entity.end,
+                    }
+                    for entity in entities
+                ]
+            }
 
         result = self.client.extract(
             self._model,
-            Item(text=text),
+            Item(text=text, metadata=metadata),
             labels=effective_labels,
         )
 

{sie_haystack-0.6.18 → sie_haystack-0.6.20}/tests/test_extractors.py

@@ -61,6 +61,25 @@ class TestSIEExtractor:
         call_kwargs = mock_sie_client.extract.call_args.kwargs
         assert call_kwargs.get("labels") == override_labels
 
+    def test_run_passes_entities_for_relation_extraction(self, mock_sie_client: object) -> None:
+        """Relation entity spans are forwarded through SIE item metadata."""
+        extractor = SIEExtractor(model="jackboyla/glirel-large-v0", labels=["ceo_of"])
+        extractor._client = mock_sie_client
+        entities = [
+            Entity(text="Tim Cook", label="PERSON", score=1.0, start=0, end=8),
+            Entity(text="Apple Inc.", label="ORG", score=1.0, start=23, end=33),
+        ]
+
+        extractor.run(text="Tim Cook is the CEO of Apple Inc.", entities=entities)
+
+        item = mock_sie_client.extract.call_args.args[1]
+        assert item["metadata"] == {
+            "entities": [
+                {"text": "Tim Cook", "label": "PERSON", "score": 1.0, "start": 0, "end": 8},
+                {"text": "Apple Inc.", "label": "ORG", "score": 1.0, "start": 23, "end": 33},
+            ]
+        }
+
     def test_custom_model(self, mock_sie_client: object) -> None:
         """Test using a custom model name."""
         extractor = SIEExtractor(model="custom/extraction-model")