qtype 0.1.12__py3-none-any.whl → 0.1.14__py3-none-any.whl

examples/research_assistant/tavily.qtype.yaml

@@ -1,289 +0,0 @@
-auths:
-- id: tavily-api_bearerauth_token
-  token: ${TAVILY-API_BEARER}
-  type: bearer_token
-description: Tools created from API specification tavily.oas.yaml
-flows: []
-id: tavily-api
-indexes: []
-memories: []
-models: []
-references: []
-tools:
-- auth: tavily-api_bearerauth_token
-  description: 'Perform a web search with various filtering and customization options.
-    Returns search results with content, URLs, scores, and optional answers. '
-  endpoint: https://api.tavily.com/search
-  headers: {}
-  id: search
-  inputs:
-    auto_parameters:
-      optional: true
-      type: boolean
-    country:
-      optional: true
-      type: text
-    days:
-      optional: true
-      type: int
-    end_date:
-      optional: true
-      type: text
-    exclude_domains:
-      optional: true
-      type:
-        element_type: text
-    include_answer:
-      optional: true
-      type: text
-    include_domains:
-      optional: true
-      type:
-        element_type: text
-    include_favicon:
-      optional: true
-      type: boolean
-    include_images:
-      optional: true
-      type: boolean
-    include_raw_content:
-      optional: true
-      type: text
-    max_results:
-      optional: true
-      type: int
-    query:
-      optional: false
-      type: text
-    search_depth:
-      optional: true
-      type: text
-    start_date:
-      optional: true
-      type: text
-    time_range:
-      optional: true
-      type: text
-    topic:
-      optional: true
-      type: text
-  method: POST
-  name: Search the web
-  outputs:
-    answer:
-      optional: true
-      type: text
-    follow_up_questions:
-      optional: true
-      type:
-        element_type: text
-    images:
-      optional: true
-      type:
-        element_type: schema_4844015520
-    query:
-      optional: true
-      type: text
-    response_time:
-      optional: true
-      type: float
-    results:
-      optional: true
-      type:
-        element_type: schema_4844016144
-  parameters: {}
-  type: APITool
-- auth: tavily-api_bearerauth_token
-  description: 'Extract and process content from one or more URLs with various formatting
-    options. Returns extracted content in text or markdown format. '
-  endpoint: https://api.tavily.com/extract
-  headers: {}
-  id: extract
-  inputs:
-    extract_depth:
-      optional: true
-      type: text
-    format:
-      optional: true
-      type: text
-    include_favicon:
-      optional: true
-      type: boolean
-    include_images:
-      optional: true
-      type: boolean
-    urls:
-      optional: false
-      type:
-        element_type: text
-  method: POST
-  name: Extract content from URLs
-  outputs:
-    failed_results:
-      optional: true
-      type:
-        element_type: schema_4844027472
-    results:
-      optional: true
-      type:
-        element_type: schema_4844024880
-  parameters: {}
-  type: APITool
-- auth: tavily-api_bearerauth_token
-  description: 'Crawl a website starting from a given URL with depth and breadth controls.
-    Extract content from discovered pages with various filtering options. '
-  endpoint: https://api.tavily.com/crawl
-  headers: {}
-  id: crawl
-  inputs:
-    allow_external:
-      optional: true
-      type: boolean
-    exclude_domains:
-      optional: true
-      type:
-        element_type: text
-    exclude_paths:
-      optional: true
-      type:
-        element_type: text
-    extract_depth:
-      optional: true
-      type: text
-    format:
-      optional: true
-      type: text
-    include_favicon:
-      optional: true
-      type: boolean
-    include_images:
-      optional: true
-      type: boolean
-    instructions:
-      optional: true
-      type: text
-    limit:
-      optional: true
-      type: int
-    max_breadth:
-      optional: true
-      type: int
-    max_depth:
-      optional: true
-      type: int
-    select_domains:
-      optional: true
-      type:
-        element_type: text
-    select_paths:
-      optional: true
-      type:
-        element_type: text
-    url:
-      optional: false
-      type: text
-  method: POST
-  name: Crawl websites
-  outputs:
-    results:
-      optional: true
-      type:
-        element_type: schema_4844408352
-  parameters: {}
-  type: APITool
-- auth: tavily-api_bearerauth_token
-  description: "Map the structure and links of a website starting from a given URL.
-    Discover the website's architecture and page hierarchy. "
-  endpoint: https://api.tavily.com/map
-  headers: {}
-  id: map
-  inputs:
-    allow_external:
-      optional: true
-      type: boolean
-    exclude_domains:
-      optional: true
-      type:
-        element_type: text
-    exclude_paths:
-      optional: true
-      type:
-        element_type: text
-    include_images:
-      optional: true
-      type: boolean
-    instructions:
-      optional: true
-      type: text
-    limit:
-      optional: true
-      type: int
-    max_breadth:
-      optional: true
-      type: int
-    max_depth:
-      optional: true
-      type: int
-    select_domains:
-      optional: true
-      type:
-        element_type: text
-    select_paths:
-      optional: true
-      type:
-        element_type: text
-    url:
-      optional: false
-      type: text
-  method: POST
-  name: Map website structure
-  outputs:
-    results:
-      optional: true
-      type:
-        element_type: schema_4844415936
-  parameters: {}
-  type: APITool
-types:
-- description: Generated from OpenAPI schema
-  id: schema_4844015520
-  properties:
-    alt_text: text?
-    title: text?
-    url: text?
-- description: Generated from OpenAPI schema
-  id: schema_4844016144
-  properties:
-    content: text?
-    favicon: text?
-    raw_content: text?
-    score: float?
-    title: text?
-    url: text?
-- description: Generated from OpenAPI schema
-  id: schema_4844024880
-  properties:
-    content: text?
-    favicon: text?
-    title: text?
-    url: text?
-- description: Generated from OpenAPI schema
-  id: schema_4844027472
-  properties:
-    error: text?
-    url: text?
-- description: Generated from OpenAPI schema
-  id: schema_4844408352
-  properties:
-    content: text?
-    favicon: text?
-    links: list[text]?
-    title: text?
-    url: text?
-- description: Generated from OpenAPI schema
-  id: schema_4844415936
-  properties:
-    depth: int?
-    links: list[text]?
-    title: text?
-    url: text?

qtype/application/facade.py

@@ -1,177 +0,0 @@
-"""Main facade for qtype operations."""
-
-from __future__ import annotations
-
-import logging
-from pathlib import Path
-from typing import Any
-
-from qtype.base.types import PathLike
-from qtype.semantic.model import Application as SemanticApplication
-from qtype.semantic.model import DocumentType as SemanticDocumentType
-
-# Note: There should be _zero_ imports here at the top that import qtype.interpreter.
-# That's the whole point of this facade - to avoid importing optional
-# dependencies unless these methods are called.
-
-logger = logging.getLogger(__name__)
-
-
-class QTypeFacade:
-    """
-    Simplified interface for qtype operations.
-
-    This facade provides lazy-loading wrappers for operations that require
-    optional dependencies (interpreter package), allowing base qtype to work
-    without those dependencies installed.
-    """
-
-    def telemetry(self, spec: SemanticDocumentType) -> None:
-        from qtype.interpreter.telemetry import register
-
-        if isinstance(spec, SemanticApplication) and spec.telemetry:
-            logger.info(
-                f"Telemetry enabled with endpoint: {spec.telemetry.endpoint}"
-            )
-            # Register telemetry if needed
-            register(spec.telemetry, self.secret_manager(spec), spec.id)
-
-    def secret_manager(self, spec: SemanticDocumentType):
-        """
-        Create a secret manager based on the specification.
-
-        Args:
-            spec: SemanticDocumentType specification
-
-        Returns:
-            Secret manager instance
-        """
-        from qtype.interpreter.base.secrets import create_secret_manager
-
-        if isinstance(spec, SemanticApplication):
-            return create_secret_manager(spec.secret_manager)
-        else:
-            raise ValueError(
-                "Can't create secret manager for non-Application spec"
-            )
-
-    async def execute_workflow(
-        self,
-        path: PathLike,
-        inputs: dict | Any,
-        flow_name: str | None = None,
-        **kwargs: Any,
-    ) -> Any:
-        """
-        Execute a complete workflow from document to results.
-
-        Args:
-            path: Path to the QType specification file
-            inputs: Dictionary of input values or DataFrame for batch
-            flow_name: Optional name of flow to execute
-            **kwargs: Additional dependencies for execution
-
-        Returns:
-            DataFrame with results (one row per input)
-        """
-        import pandas as pd
-        from opentelemetry import trace
-
-        from qtype.interpreter.base.executor_context import ExecutorContext
-        from qtype.interpreter.converters import (
-            dataframe_to_flow_messages,
-            flow_messages_to_dataframe,
-        )
-        from qtype.interpreter.flow import run_flow
-        from qtype.interpreter.types import Session
-        from qtype.semantic.loader import load
-
-        # Load the semantic application
-        semantic_model, type_registry = load(Path(path))
-        assert isinstance(semantic_model, SemanticApplication)
-        self.telemetry(semantic_model)
-
-        # Find the flow to execute
-        if flow_name:
-            target_flow = None
-            for flow in semantic_model.flows:
-                if flow.id == flow_name:
-                    target_flow = flow
-                    break
-            if target_flow is None:
-                raise ValueError(f"Flow '{flow_name}' not found")
-        else:
-            if semantic_model.flows:
-                target_flow = semantic_model.flows[0]
-            else:
-                raise ValueError("No flows found in application")
-
-        logger.info(f"Executing flow {target_flow.id} from {path}")
-
-        # Convert inputs to DataFrame (normalize single dict to 1-row DataFrame)
-
-        if isinstance(inputs, dict):
-            input_df = pd.DataFrame([inputs])
-        elif isinstance(inputs, pd.DataFrame):
-            input_df = inputs
-        else:
-            raise ValueError(
-                f"Inputs must be dict or DataFrame, got {type(inputs)}"
-            )
-
-        # Create session
-        session = Session(
-            session_id=kwargs.pop("session_id", "default"),
-            conversation_history=kwargs.pop("conversation_history", []),
-        )
-
-        # Convert DataFrame to FlowMessages
-        initial_messages = dataframe_to_flow_messages(input_df, session)
-
-        # Execute the flow
-        secret_manager = self.secret_manager(semantic_model)
-
-        context = ExecutorContext(
-            secret_manager=secret_manager,
-            tracer=trace.get_tracer(__name__),
-        )
-        results = await run_flow(
-            target_flow,
-            initial_messages,
-            context=context,
-            **kwargs,
-        )
-
-        # Convert results back to DataFrame
-        results_df = flow_messages_to_dataframe(results, target_flow)
-
-        return results_df
-
-    def generate_aws_bedrock_models(self) -> list[dict[str, Any]]:
-        """
-        Generate AWS Bedrock model definitions.
-
-        Returns:
-            List of model definitions for AWS Bedrock models.
-
-        Raises:
-            ImportError: If boto3 is not installed.
-            Exception: If AWS API call fails.
-        """
-        import boto3  # type: ignore[import-untyped]
-
-        logger.info("Discovering AWS Bedrock models...")
-        client = boto3.client("bedrock")
-        models = client.list_foundation_models()
-
-        model_definitions = []
-        for model_summary in models.get("modelSummaries", []):
-            model_definitions.append(
-                {
-                    "id": model_summary["modelId"],
-                    "provider": "aws-bedrock",
-                }
-            )
-
-        logger.info(f"Discovered {len(model_definitions)} AWS Bedrock models")
-        return model_definitions