qtype 0.1.11__py3-none-any.whl → 0.1.13__py3-none-any.whl

qtype/semantic/checker.py

@@ -50,6 +50,10 @@ class FlowHasNoStepsError(QTypeValidationError):
 # Alias for backward compatibility and semantic clarity
 QTypeSemanticError = SemanticError
 
+# Step types that support text streaming
+# These are the only step types that can produce streaming text output
+STREAMING_STEP_TYPES = (LLMInference, Agent)
+
 
 # ---- Helper Functions for Common Validation Patterns ----
 
@@ -543,6 +547,21 @@ def _validate_flow(flow: Flow) -> None:
                     f"Flow {flow.id} has a Complete interface but {len(text_outputs)} text outputs -- there should be 1."
                 )
 
+            # Ensure the final step supports streaming for Complete interface
+            if flow.steps:
+                final_step = flow.steps[-1]
+                if not isinstance(final_step, STREAMING_STEP_TYPES):
+                    streaming_type_names = ", ".join(
+                        t.__name__ for t in STREAMING_STEP_TYPES
+                    )
+                    raise QTypeSemanticError(
+                        (
+                            f"Flow {flow.id} has a Complete interface which requires streaming output, "
+                            f"but the final step '{final_step.id}' is of type '{final_step.type}' which does not support streaming. "
+                            f"The final step must be one of: {streaming_type_names}."
+                        )
+                    )
+
 
 def _has_secret_reference(obj: Any) -> bool:
     """
@@ -704,7 +723,7 @@ def check(model: BaseModel) -> None:
     # Check if this model type has a validator
     model_type = type(model)
     if model_type in _VALIDATORS:
-        _VALIDATORS[model_type](model)
+        _VALIDATORS[model_type](model)  # type: ignore[arg-type]
 
     # Recursively validate all fields
     for field_name, field_value in model:

qtype/semantic/generate.py

@@ -73,7 +73,7 @@ def sort_classes_by_inheritance(
             ):
                 graph.add_edge(base.__name__, class_name)
 
-    sorted_names = list(nx.topological_sort(graph))
+    sorted_names = list(nx.topological_sort(graph))  # type: ignore[arg-type]
 
     # sorted_names = sorted(graph.nodes, key=lambda node: depths[node])
     return [(name, class_dict[name]) for name in sorted_names]
@@ -144,8 +144,7 @@ def generate_semantic_model(args: argparse.Namespace) -> None:
                 CustomType,
                 DecoderFormat,
                 ListType,
-                PrimitiveTypeEnum,
-                ToolParameter
+                PrimitiveTypeEnum
             )
             from qtype.dsl.model import Variable as DSLVariable  # noqa: F401
             from qtype.dsl.model import VariableType  # noqa: F401
@@ -158,10 +157,8 @@ def generate_semantic_model(args: argparse.Namespace) -> None:
         f.write(
             dedent('''
             class Variable(DSLVariable, BaseModel):
-                """Semantic version of DSL Variable with ID references resolved."""
-                value: Any | None = Field(None, description="The value of the variable")
-                def is_set(self) -> bool:
-                    return self.value is not None
+                """Semantic version of DSL Variable."""
+                pass
 
         ''').lstrip()
         )
@@ -490,10 +487,10 @@ def generate_semantic_class(class_name: str, cls: type) -> str:
         # Only process fields that are actually defined on this class
         for field_name in cls.__annotations__:
             if (
-                field_name in cls.model_fields
+                field_name in cls.model_fields  # type: ignore[operator]
                 and f"{class_name}.{field_name}" not in FIELDS_TO_IGNORE
             ):
-                field_info = cls.model_fields[field_name]
+                field_info = cls.model_fields[field_name]  # type: ignore[index]
                 field_type = field_info.annotation
                 field_default = field_info.default
                 field_default_factory = field_info.default_factory

qtype/semantic/model.py

@@ -32,19 +32,15 @@ from qtype.dsl.model import (  # noqa: F401
     DecoderFormat,
     ListType,
     PrimitiveTypeEnum,
-    ToolParameter,
 )
 from qtype.dsl.model import Variable as DSLVariable  # noqa: F401
 from qtype.semantic.base_types import ImmutableModel
 
 
 class Variable(DSLVariable, BaseModel):
-    """Semantic version of DSL Variable with ID references resolved."""
+    """Semantic version of DSL Variable."""
 
-    value: Any | None = Field(None, description="The value of the variable")
-
-    def is_set(self) -> bool:
-        return self.value is not None
+    pass
 
 
 class AuthorizationProvider(ImmutableModel):
@@ -66,12 +62,12 @@ class Tool(ImmutableModel):
     description: str = Field(
         ..., description="Description of what the tool does."
     )
-    inputs: dict[str, ToolParameter] = Field(
-        default_factory=dict,
+    inputs: list[Variable] = Field(
+        default_factory=list,
         description="Input parameters required by this tool.",
     )
-    outputs: dict[str, ToolParameter] = Field(
-        default_factory=dict,
+    outputs: list[Variable] = Field(
+        default_factory=list,
         description="Output parameters produced by this tool.",
     )
 
@@ -421,9 +417,9 @@ class APITool(Tool):
         default_factory=dict,
         description="Optional HTTP headers to include in the request.",
     )
-    parameters: dict[str, ToolParameter] = Field(
-        default_factory=dict,
-        description="Output parameters produced by this tool.",
+    parameters: list[Variable] = Field(
+        default_factory=list,
+        description="Path and query parameters for the API call.",
     )
 
 
@@ -447,15 +443,12 @@ class AWSSecretManager(SecretManager):
 
 class Aggregate(Step):
     """
-    A terminal step that consumes an entire input stream and produces a single
-    summary message with success/error counts.
+    A step that, after all messages have been processed,
+    returns a single message containing the counts of successful and failed
+    messages. Other messages are passed through unchanged.
     """
 
     type: Literal["Aggregate"] = Field("Aggregate")
-    outputs: list[Variable] = Field(
-        default_factory=list,
-        description="References to the variables for the output. There should be one and only one output with type AggregateStats",
-    )
 
 
 class Collect(Step, BatchableStepMixin):
@@ -472,8 +465,9 @@ class Construct(Step):
     """A step that converts variables into an instance of a Custom or Domain Type"""
 
     type: Literal["Construct"] = Field("Construct")
-    field_mapping: dict[str, str] = Field(
-        ..., description="Mapping of type inputs to variable names, if needed."
+    field_bindings: dict[str, Variable] = Field(
+        ...,
+        description="Mapping from type field names to flow variable names.",
     )
 
 
@@ -558,6 +552,9 @@ class FieldExtractor(Step):
     The extracted data is used to construct the output variable by passing it
     as keyword arguments to the output type's constructor.
 
+    If there is no match and the output variable is optional, it is set to None.
+    If there is no match and the output variable is required, an error is raised.
+
     Example JSONPath expressions:
     - `$.field_name` - Extract a single field
     - `$.items[*]` - Extract all items from a list
@@ -569,10 +566,6 @@ class FieldExtractor(Step):
         ...,
         description="JSONPath expression to extract data from the input. Uses jsonpath-ng syntax.",
     )
-    fail_on_missing: bool = Field(
-        True,
-        description="Whether to raise an error if the JSONPath matches no data. If False, returns None.",
-    )
 
 
 class InvokeEmbedding(Step, ConcurrentStepMixin):
@@ -590,13 +583,13 @@ class InvokeFlow(Step):
 
     type: Literal["InvokeFlow"] = Field("InvokeFlow")
     flow: Flow = Field(..., description="Flow to invoke.")
-    input_bindings: dict[Variable, str] = Field(
+    input_bindings: dict[str, Variable] = Field(
         ...,
-        description="Mapping from variable references to flow input variable IDs.",
+        description="Mapping from flow input variable IDs to step variable names.",
     )
-    output_bindings: dict[Variable, str] = Field(
+    output_bindings: dict[str, Variable] = Field(
         ...,
-        description="Mapping from variable references to flow output variable IDs.",
+        description="Mapping from flow output variable IDs to step variable names.",
     )
 
 
@@ -605,13 +598,13 @@ class InvokeTool(Step, ConcurrentStepMixin):
 
     type: Literal["InvokeTool"] = Field("InvokeTool")
     tool: Tool = Field(..., description="Tool to invoke.")
-    input_bindings: dict[str, str] = Field(
+    input_bindings: dict[str, Variable] = Field(
         ...,
-        description="Mapping from variable references to tool input parameter names.",
+        description="Mapping from tool parameter names to flow variable names.",
     )
-    output_bindings: dict[str, str] = Field(
+    output_bindings: dict[str, Variable] = Field(
         ...,
-        description="Mapping from variable references to tool output parameter names.",
+        description="Mapping from tool output names to flow variable names.",
     )
 
 

qtype/semantic/resolver.py

@@ -71,6 +71,13 @@ def to_semantic_ir(
         # If the object is already in the symbol table, return it.
         return symbol_table[obj_id]
 
+    if isinstance(dslobj, dict):
+        # If the object is a dict, recursively resolve each value
+        return {
+            key: to_semantic_ir(value, symbol_table)
+            for key, value in dslobj.items()
+        }
+
     if isinstance(dslobj, list):
         # If the object is a list, we will resolve each item in the list.
         return [to_semantic_ir(item, symbol_table) for item in dslobj]  # type: ignore

qtype/semantic/visualize.py

@@ -21,6 +21,8 @@ from qtype.semantic.model import (
     DocumentIndex,
     DocumentSearch,
     Flow,
+    InvokeFlow,
+    InvokeTool,
     LLMInference,
     Memory,
     Model,
@@ -47,7 +49,7 @@ def visualize_application(app: Application) -> str:
     """
     lines = [
         "flowchart TD",
-        f'    subgraph APP ["📱 Application: {app.id}"]',
+        f'    subgraph APP ["📱 {app.id}"]',
         "        direction TB",
         "",
     ]
@@ -117,8 +119,8 @@ def _generate_flow_subgraph(
     flow: Flow, flow_id: str
 ) -> tuple[list[str], list[str]]:
     """Generate a flow subgraph with internal nodes and return external connections."""
-    # Add more spacing and line breaks for better SVG rendering
-    description = f"\n{flow.description}" if flow.description else ""
+    # Keep labels concise - no multi-line descriptions in labels
+    flow_label = f"🔄 {flow.id}"
 
     # Choose direction based on flow characteristics:
     # - Flows with interface (e.g., Conversational) use LR (left-right)
@@ -126,7 +128,7 @@ def _generate_flow_subgraph(
     direction = "LR" if flow.interface else "TB"
 
     lines = [
-        f'    subgraph {flow_id} ["🔄 Flow: {flow.id}{description}"]',
+        f'    subgraph {flow_id} ["{flow_label}"]',
         f"        direction {direction}",
     ]
 
@@ -167,20 +169,31 @@ def _generate_step_node(
     lines = []
     external_connections = []
 
-    if isinstance(step, Flow):
-        # Nested flow
+    if isinstance(step, InvokeFlow):
         lines.append(
-            f'        {node_id}@{{shape: subproc, label: "📋 Flow: {step.id}"}}'
+            f'        {node_id}@{{shape: sub-r, label: "🔄 {step.id}"}}'
+        )
+        # Connect to the invoked flow
+        invoked_flow_id = f"FLOW_{_sanitize_id(step.flow.id)}"
+        external_connections.append(
+            f"    {node_id} -.->|invokes| {invoked_flow_id}"
         )
     elif isinstance(step, Agent):
         # Agent with tools
         lines.append(
-            f'        {node_id}@{{shape: hex, label: "🤖 Agent: {step.id}"}}'
+            f'        {node_id}@{{shape: hex, label: "🤖 {step.id}"}}'
         )
         # Connect to tools
         for tool in step.tools:
             tool_id = f"TOOL_{_sanitize_id(tool.id)}"
-            external_connections.append(f"    {node_id} -.-> {tool_id}")
+            external_connections.append(f"    {node_id} -.->|uses| {tool_id}")
+    elif isinstance(step, InvokeTool):
+        lines.append(
+            f'        {node_id}@{{shape: rect, label: "⚙️ {step.id}"}}'
+        )
+
+        tool_id = f"TOOL_{_sanitize_id(step.tool.id)}"
+        external_connections.append(f"    {node_id} -.->|uses| {tool_id}")
     elif isinstance(step, LLMInference):
         lines.append(
             f'        {node_id}@{{shape: rounded, label: "✨ {step.id}"}}'
@@ -196,7 +209,7 @@ def _generate_step_node(
             )
     elif isinstance(step, PromptTemplate):
         lines.append(
-            f'        {node_id}@{{shape: doc, label: "📄 Template: {step.id}"}}'
+            f'        {node_id}@{{shape: doc, label: "📄 {step.id}"}}'
         )
     elif isinstance(step, Decoder):
         format_label = (
@@ -205,46 +218,30 @@ def _generate_step_node(
             else str(step.format)
         )
         lines.append(
-            f'        {node_id}@{{shape: lean-r, label: "🔍 Decode: {step.id} ({format_label})"}}'
+            f'        {node_id}@{{shape: lean-r, label: "🔍 {step.id} ({format_label})"}}'
         )
     elif isinstance(step, VectorSearch):
         lines.append(
-            f'        {node_id}@{{shape: cyl, label: "🔎 Vector Search: {step.id}"}}'
+            f'        {node_id}@{{shape: cyl, label: "🔎 {step.id}"}}'
         )
         index_id = f"INDEX_{_sanitize_id(step.index.id)}"
         external_connections.append(f"    {node_id} -.-> {index_id}")
     elif isinstance(step, DocumentSearch):
         lines.append(
-            f'        {node_id}@{{shape: cyl, label: "📚 Doc Search: {step.id}"}}'
+            f'        {node_id}@{{shape: cyl, label: "📚 {step.id}"}}'
         )
         index_id = f"INDEX_{_sanitize_id(step.index.id)}"
         external_connections.append(f"    {node_id} -.-> {index_id}")
     elif isinstance(step, Search):
         lines.append(
-            f'        {node_id}@{{shape: cyl, label: "🔍 Search: {step.id}"}}'
+            f'        {node_id}@{{shape: cyl, label: "🔍 {step.id}"}}'
         )
         index_id = f"INDEX_{_sanitize_id(step.index.id)}"
         external_connections.append(f"    {node_id} -.-> {index_id}")
-    elif isinstance(step, APITool):
-        method_label = step.method.upper()
-        lines.append(
-            f'        {node_id}["⚡ API: {step.id} ({method_label})"]'
-        )
-        if step.auth:
-            auth_id = f"AUTH_{_sanitize_id(step.auth.id)}"
-            external_connections.append(f"    {node_id} -.-> {auth_id}")
-    elif isinstance(step, PythonFunctionTool):
-        lines.append(
-            f'        {node_id}@{{shape: rect, label: "🐍 Python: {step.id} {step.function_name}"}}'
-        )
-    elif isinstance(step, Tool):
-        lines.append(
-            f'        {node_id}@{{shape: rect, label: "🔧 Tool: {step.id}"}}'
-        )
     else:
         # Generic step
         lines.append(
-            f'        {node_id}@{{shape: rect, label: "⚙️ Step: {step.id}"}}'
+            f'        {node_id}@{{shape: rect, label: "⚙️ {step.id}"}}'
         )
 
     return lines, external_connections
@@ -276,20 +273,11 @@ def _generate_step_connections(
                 # consumes and produces the same variable)
                 if producer_id == node_id:
                     continue
-                # Get a simple string representation of the variable type
-                var_type = str(input_var.type).split(".")[
-                    -1
-                ]  # Get the last part after dots
-                var_id_and_type = f"{input_var.id}: {var_type}"
-
-                is_list = str(var_type).startswith("list[") and str(
-                    var_type
-                ).endswith("]")
-                if is_list:
-                    var_id_and_type = f'"{var_id_and_type}"'
+                # Use simple variable name only - no type annotations
+                var_label = input_var.id
 
                 lines.append(
-                    f"        {producer_id} -->|{var_id_and_type}| {node_id}"
+                    f"        {producer_id} -->|{var_label}| {node_id}"
                 )
 
         # Then, register this step's outputs for future steps
@@ -374,7 +362,7 @@ def _generate_shared_resources(app: Application) -> list[str]:
             auth_id = f"AUTH_{_sanitize_id(auth.id)}"
             auth_type = auth.type.upper()
             lines.append(
-                f'        {auth_id}@{{shape: hex, label: "🔐 {auth.id}\\n{auth_type}"}}'
+                f'        {auth_id}@{{shape: hex, label: "🔐 {auth.id} ({auth_type})"}}'
             )
 
         # Models
@@ -394,7 +382,7 @@ def _generate_shared_resources(app: Application) -> list[str]:
             index_id = f"INDEX_{_sanitize_id(index.id)}"
             if isinstance(index, VectorIndex):
                 lines.append(
-                    f'        {index_id}@{{shape: cyl, label: "🗂️ Vector: {index.id}"}}'
+                    f'        {index_id}@{{shape: cyl, label: "🗂️ {index.id}"}}'
                 )
                 # Connect to embedding model
                 emb_model_id = f"EMB_{_sanitize_id(index.embedding_model.id)}"
@@ -404,7 +392,7 @@ def _generate_shared_resources(app: Application) -> list[str]:
                 lines.append(f"        {index_id} -.->|embeds| {emb_model_id}")
             elif isinstance(index, DocumentIndex):
                 lines.append(
-                    f'        {index_id}@{{shape: cyl, label: "📚 Docs: {index.id}"}}'
+                    f'        {index_id}@{{shape: cyl, label: "📚 {index.id}"}}'
                 )
             else:
                 lines.append(
@@ -412,11 +400,15 @@ def _generate_shared_resources(app: Application) -> list[str]:
                 )
 
             if index.auth:
-                # Handle auth as either AuthorizationProvider object or string ID
-                if isinstance(index.auth, str):
-                    auth_id = f"AUTH_{_sanitize_id(index.auth)}"
+                auth_value = index.auth
+                if isinstance(auth_value, str):
+                    auth_ref = auth_value
                 else:
-                    auth_id = f"AUTH_{_sanitize_id(index.auth.id)}"
+                    auth_ref = getattr(auth_value, "id", None)
+                    if auth_ref is None:
+                        auth_ref = str(auth_value)
+
+                auth_id = f"AUTH_{_sanitize_id(str(auth_ref))}"
                 lines.append(f"        {index_id} -.->|uses| {auth_id}")
 
         # Memories
@@ -428,7 +420,7 @@ def _generate_shared_resources(app: Application) -> list[str]:
                 else str(memory.token_limit)
             )
             lines.append(
-                f'        {memory_id}@{{shape: win-pane, label: "🧠 {memory.id}\\n{token_limit}T"}}'
+                f'        {memory_id}@{{shape: win-pane, label: "🧠 {memory.id} ({token_limit}T)"}}'
             )
 
         # Tools (if not already covered by flows)
@@ -444,7 +436,7 @@ def _generate_shared_resources(app: Application) -> list[str]:
                     lines.append(f"        {tool_id} -.->|uses| {auth_id}")
             elif isinstance(tool, PythonFunctionTool):
                 lines.append(
-                    f'        {tool_id}@{{shape: rect, label: "🐍 {tool.id}\\n{tool.function_name}"}}'
+                    f'        {tool_id}@{{shape: rect, label: "🐍 {tool.id}"}}'
                 )
             else:
                 lines.append(
@@ -459,7 +451,7 @@ def _generate_shared_resources(app: Application) -> list[str]:
 def _generate_telemetry_nodes(telemetry: TelemetrySink) -> list[str]:
     """Generate nodes for telemetry configuration."""
     # Replace :// with a space to avoid markdown link parsing
-    safe_endpoint = telemetry.endpoint.replace("://", "://")
+    safe_endpoint = telemetry.endpoint.replace("://", "://")  # type: ignore[union-attr]
 
     lines = [
         '    subgraph TELEMETRY ["📊 Observability"]',

{qtype-0.1.11.dist-info → qtype-0.1.13.dist-info}/METADATA

@@ -1,57 +1,61 @@
 Metadata-Version: 2.4
 Name: qtype
-Version: 0.1.11
+Version: 0.1.13
 Summary: DSL for Generative AI Prototyping
+Project-URL: Homepage, https://github.com/bazaarvoice/qtype
 Author-email: Lou Kratz <lou.kratz+qtype@bazaarvoice.com>
 License-Expression: Apache-2.0
-Project-URL: Homepage, https://github.com/bazaarvoice/qtype
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: jsonschema>=4.24.0
-Requires-Dist: pydantic>=2.12.4
-Requires-Dist: pyyaml>=6.0.2
-Requires-Dist: python-dotenv>=1.0.0
-Requires-Dist: openai>=1.93.0
+Requires-Python: >=3.10
 Requires-Dist: fsspec>=2025.5.1
+Requires-Dist: google-cloud-aiplatform>=1.120.0
+Requires-Dist: jsonschema>=4.24.0
 Requires-Dist: mkdocs-awesome-pages-plugin>=2.10.1
-Requires-Dist: pip-system-certs>=5.2
+Requires-Dist: openai>=1.93.0
 Requires-Dist: openapi3-parser>=1.1.21
+Requires-Dist: pip-system-certs>=5.2
 Requires-Dist: pydantic-yaml>=1.6.0
-Requires-Dist: google-cloud-aiplatform>=1.120.0
+Requires-Dist: pydantic>=2.12.4
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: pyyaml>=6.0.2
 Provides-Extra: interpreter
-Requires-Dist: aiostream>=0.7.1; extra == "interpreter"
-Requires-Dist: arize-phoenix-otel>=0.12.1; extra == "interpreter"
-Requires-Dist: boto3>=1.34.0; extra == "interpreter"
-Requires-Dist: datasets>=4.4.1; extra == "interpreter"
-Requires-Dist: diskcache>=5.6.3; extra == "interpreter"
-Requires-Dist: docling>=2.55.1; extra == "interpreter"
-Requires-Dist: docx2txt>=0.9; extra == "interpreter"
-Requires-Dist: fastapi>=0.116.1; extra == "interpreter"
-Requires-Dist: jsonpath-ng>=1.7.0; extra == "interpreter"
-Requires-Dist: langfuse>=3.9.0; extra == "interpreter"
-Requires-Dist: llama-index-embeddings-bedrock>=0.5.2; extra == "interpreter"
-Requires-Dist: llama-index-embeddings-openai>=0.3.1; extra == "interpreter"
-Requires-Dist: llama-index-llms-bedrock-converse>=0.10.5; extra == "interpreter"
-Requires-Dist: llama-index-llms-bedrock>=0.3.8; extra == "interpreter"
-Requires-Dist: llama-index-llms-vertex>=0.6.1; extra == "interpreter"
-Requires-Dist: llama-index-postprocessor-bedrock-rerank>=0.5.1; extra == "interpreter"
-Requires-Dist: llama-index-readers-huggingface-fs>=0.4.1; extra == "interpreter"
-Requires-Dist: llama-index-vector-stores-qdrant>=0.8.6; extra == "interpreter"
-Requires-Dist: llama-index>=0.12.45; extra == "interpreter"
-Requires-Dist: openinference-instrumentation-llama-index>=4.3.4; extra == "interpreter"
-Requires-Dist: opensearch-py>=2.7.0; extra == "interpreter"
-Requires-Dist: opentelemetry-exporter-otlp>=1.35.0; extra == "interpreter"
-Requires-Dist: opentelemetry-sdk>=1.35.0; extra == "interpreter"
-Requires-Dist: pandas>=2.2.3; extra == "interpreter"
-Requires-Dist: psycopg2-binary>=2.9.10; extra == "interpreter"
-Requires-Dist: pyarrow>=21.0.0; extra == "interpreter"
-Requires-Dist: pyathena[sqlalchemy]>=3.18.0; extra == "interpreter"
-Requires-Dist: python-magic>=0.4.27; extra == "interpreter"
-Requires-Dist: s3fs>=2025.7.0; extra == "interpreter"
-Requires-Dist: sqlalchemy>=2.0.42; extra == "interpreter"
-Requires-Dist: uvicorn[standard]>=0.35.0; extra == "interpreter"
-Dynamic: license-file
+Requires-Dist: aiostream>=0.7.1; extra == 'interpreter'
+Requires-Dist: arize-phoenix-otel>=0.12.1; extra == 'interpreter'
+Requires-Dist: boto3>=1.34.0; extra == 'interpreter'
+Requires-Dist: datasets>=4.4.1; extra == 'interpreter'
+Requires-Dist: diskcache>=5.6.3; extra == 'interpreter'
+Requires-Dist: docling>=2.55.1; extra == 'interpreter'
+Requires-Dist: docx2txt>=0.9; extra == 'interpreter'
+Requires-Dist: fastapi>=0.116.1; extra == 'interpreter'
+Requires-Dist: jsonpath-ng>=1.7.0; extra == 'interpreter'
+Requires-Dist: langfuse>=3.9.0; extra == 'interpreter'
+Requires-Dist: llama-index-embeddings-bedrock>=0.5.2; extra == 'interpreter'
+Requires-Dist: llama-index-embeddings-openai>=0.3.1; extra == 'interpreter'
+Requires-Dist: llama-index-llms-bedrock-converse>=0.10.5; extra == 'interpreter'
+Requires-Dist: llama-index-llms-bedrock>=0.3.8; extra == 'interpreter'
+Requires-Dist: llama-index-llms-vertex>=0.6.1; extra == 'interpreter'
+Requires-Dist: llama-index-postprocessor-bedrock-rerank>=0.5.1; extra == 'interpreter'
+Requires-Dist: llama-index-readers-huggingface-fs>=0.4.1; extra == 'interpreter'
+Requires-Dist: llama-index-vector-stores-qdrant>=0.8.6; extra == 'interpreter'
+Requires-Dist: llama-index>=0.12.45; extra == 'interpreter'
+Requires-Dist: openinference-instrumentation-llama-index>=4.3.4; extra == 'interpreter'
+Requires-Dist: opensearch-py>=2.7.0; extra == 'interpreter'
+Requires-Dist: opentelemetry-exporter-otlp>=1.35.0; extra == 'interpreter'
+Requires-Dist: opentelemetry-sdk>=1.35.0; extra == 'interpreter'
+Requires-Dist: pandas>=2.2.3; extra == 'interpreter'
+Requires-Dist: psycopg2-binary>=2.9.10; extra == 'interpreter'
+Requires-Dist: pyarrow>=21.0.0; extra == 'interpreter'
+Requires-Dist: pyathena[sqlalchemy]>=3.18.0; extra == 'interpreter'
+Requires-Dist: python-magic>=0.4.27; extra == 'interpreter'
+Requires-Dist: s3fs>=2025.7.0; extra == 'interpreter'
+Requires-Dist: sqlalchemy>=2.0.42; extra == 'interpreter'
+Requires-Dist: uvicorn[standard]>=0.35.0; extra == 'interpreter'
+Provides-Extra: mcp
+Requires-Dist: cachetools>=6.2.1; extra == 'mcp'
+Requires-Dist: httpx>=0.28.1; extra == 'mcp'
+Requires-Dist: mcp[cli]>=1.25.0; extra == 'mcp'
+Requires-Dist: tantivy>=0.25.1; extra == 'mcp'
+Description-Content-Type: text/markdown
 
 # QType
 
@@ -130,6 +134,23 @@ And go to [http://localhost:8000/ui](http://localhost:8000/ui) to see the user i
 
 See the [full docs](https://bazaarvoice.github.io/qtype/) for more examples and guides.
 
+## ✨ Developing with AI?
+
+Use the QType MCP server to speed yourself up! Just set your assistant to run `qtype mcp`.
+For VSCode, just add the following to `.vscode/mcp.json`:
+
+```json
+{
+  "servers": {
+    "qtype": {
+      "type": "stdio",
+      "command": "qtype",
+      "cwd": "${workspaceFolder}",
+      "args": ["mcp", "--transport", "stdio"]
+    }
+  }
+}
+```
 
 
 ## 🤝 Contributing
@@ -158,4 +179,4 @@ Stay tuned for upcoming features like:
 Happy hacking with QType! 🛠️
 
 
-[![Generate JSON Schema](https://github.com/bazaarvoice/qtype/actions/workflows/github_workflows_generate-schema.yml/badge.svg)](https://github.com/bazaarvoice/qtype/actions/workflows/github_workflows_generate-schema.yml) [![Publish to PyPI](https://github.com/bazaarvoice/qtype/actions/workflows/publish-pypi.yml/badge.svg)](https://github.com/bazaarvoice/qtype/actions/workflows/publish-pypi.yml)
+[![Generate JSON Schema](https://github.com/bazaarvoice/qtype/actions/workflows/github_workflows_generate-schema.yml/badge.svg)](https://github.com/bazaarvoice/qtype/actions/workflows/github_workflows_generate-schema.yml) [![Publish to PyPI](https://github.com/bazaarvoice/qtype/actions/workflows/publish-pypi.yml/badge.svg)](https://github.com/bazaarvoice/qtype/actions/workflows/publish-pypi.yml)