causaliq-knowledge 0.2.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

causaliq_knowledge/cli.py

@@ -1,414 +0,0 @@
-"""Command-line interface for causaliq-knowledge."""
-
-from __future__ import annotations
-
-import json
-import sys
-from typing import Optional
-
-import click
-
-from causaliq_knowledge import __version__
-
-
-@click.group()
-@click.version_option(version=__version__)
-def cli() -> None:
-    """CausalIQ Knowledge - LLM knowledge for causal discovery.
-
-    Query LLMs about causal relationships between variables.
-    """
-    pass
-
-
-@cli.command("query")
-@click.argument("node_a")
-@click.argument("node_b")
-@click.option(
-    "--model",
-    "-m",
-    multiple=True,
-    default=["groq/llama-3.1-8b-instant"],
-    help="LLM model(s) to query. Can be specified multiple times.",
-)
-@click.option(
-    "--domain",
-    "-d",
-    default=None,
-    help="Domain context (e.g., 'medicine', 'economics').",
-)
-@click.option(
-    "--strategy",
-    "-s",
-    type=click.Choice(["weighted_vote", "highest_confidence"]),
-    default="weighted_vote",
-    help="Consensus strategy for multi-model queries.",
-)
-@click.option(
-    "--json",
-    "output_json",
-    is_flag=True,
-    help="Output result as JSON.",
-)
-@click.option(
-    "--temperature",
-    "-t",
-    type=float,
-    default=0.1,
-    help="LLM temperature (0.0-1.0).",
-)
-def query_edge(
-    node_a: str,
-    node_b: str,
-    model: tuple[str, ...],
-    domain: Optional[str],
-    strategy: str,
-    output_json: bool,
-    temperature: float,
-) -> None:
-    """Query LLMs about a causal relationship between two variables.
-
-    NODE_A and NODE_B are the variable names to query about.
-
-    Examples:
-
-        cqknow query smoking lung_cancer
-
-        cqknow query smoking lung_cancer --domain medicine
-
-        cqknow query X Y --model groq/llama-3.1-8b-instant \
-                         --model gemini/gemini-2.5-flash
-    """
-    # Import here to avoid slow startup for --help
-    from causaliq_knowledge.llm import LLMKnowledge
-
-    # Build context
-    context = None
-    if domain:
-        context = {"domain": domain}
-
-    # Create provider
-    try:
-        provider = LLMKnowledge(
-            models=list(model),
-            consensus_strategy=strategy,
-            temperature=temperature,
-        )
-    except Exception as e:
-        click.echo(f"Error creating provider: {e}", err=True)
-        sys.exit(1)
-
-    # Query
-    click.echo(
-        f"Querying {len(model)} model(s) about: {node_a} -> {node_b}",
-        err=True,
-    )
-
-    try:
-        result = provider.query_edge(node_a, node_b, context=context)
-    except Exception as e:
-        click.echo(f"Error querying LLM: {e}", err=True)
-        sys.exit(1)
-
-    # Output
-    if output_json:
-        output = {
-            "node_a": node_a,
-            "node_b": node_b,
-            "exists": result.exists,
-            "direction": result.direction.value if result.direction else None,
-            "confidence": result.confidence,
-            "reasoning": result.reasoning,
-            "model": result.model,
-        }
-        click.echo(json.dumps(output, indent=2))
-    else:
-        # Human-readable output
-        exists_map = {True: "Yes", False: "No", None: "Uncertain"}
-        exists_str = exists_map[result.exists]
-        direction_str = result.direction.value if result.direction else "N/A"
-
-        click.echo(f"\n{'='*60}")
-        click.echo(f"Query: Does '{node_a}' cause '{node_b}'?")
-        click.echo("=" * 60)
-        click.echo(f"Exists:     {exists_str}")
-        click.echo(f"Direction:  {direction_str}")
-        click.echo(f"Confidence: {result.confidence:.2f}")
-        click.echo(f"Model(s):   {result.model or 'unknown'}")
-        click.echo(f"{'='*60}")
-        click.echo(f"Reasoning:  {result.reasoning}")
-        click.echo()
-
-    # Show stats
-    stats = provider.get_stats()
-    if stats["total_cost"] > 0:
-        click.echo(
-            f"Cost: ${stats['total_cost']:.6f} "
-            f"({stats['total_calls']} call(s))",
-            err=True,
-        )
-
-
-@cli.command("models")
-@click.argument("provider", required=False, default=None)
-def list_models(provider: Optional[str]) -> None:
-    """List available LLM models from each provider.
-
-    Queries each provider's API to show models accessible with your
-    current configuration. Results are filtered by your API key's
-    access level or locally installed models.
-
-    Optionally specify PROVIDER to list models from a single provider:
-    groq, anthropic, gemini, ollama, openai, deepseek, or mistral.
-
-    Examples:
-
-        cqknow models              # List all providers
-
-        cqknow models groq         # List only Groq models
-
-        cqknow models mistral      # List only Mistral models
-    """
-    from typing import Callable, List, Optional, Tuple, TypedDict
-
-    from causaliq_knowledge.llm import (
-        AnthropicClient,
-        AnthropicConfig,
-        DeepSeekClient,
-        DeepSeekConfig,
-        GeminiClient,
-        GeminiConfig,
-        GroqClient,
-        GroqConfig,
-        MistralClient,
-        MistralConfig,
-        OllamaClient,
-        OllamaConfig,
-        OpenAIClient,
-        OpenAIConfig,
-    )
-
-    # Type for get_models functions
-    GetModelsFunc = Callable[[], Tuple[bool, List[str], Optional[str]]]
-
-    class ProviderInfo(TypedDict):
-        name: str
-        prefix: str
-        env_var: Optional[str]
-        url: str
-        get_models: GetModelsFunc
-
-    def get_groq_models() -> Tuple[bool, List[str], Optional[str]]:
-        """Returns (available, models, error_msg)."""
-        try:
-            client = GroqClient(GroqConfig())
-            if not client.is_available():
-                return False, [], "GROQ_API_KEY not set"
-            models = [f"groq/{m}" for m in client.list_models()]
-            return True, models, None
-        except ValueError as e:
-            return False, [], str(e)
-
-    def get_anthropic_models() -> Tuple[bool, List[str], Optional[str]]:
-        """Returns (available, models, error_msg)."""
-        try:
-            client = AnthropicClient(AnthropicConfig())
-            if not client.is_available():
-                return False, [], "ANTHROPIC_API_KEY not set"
-            models = [f"anthropic/{m}" for m in client.list_models()]
-            return True, models, None
-        except ValueError as e:
-            return False, [], str(e)
-
-    def get_gemini_models() -> Tuple[bool, List[str], Optional[str]]:
-        """Returns (available, models, error_msg)."""
-        try:
-            client = GeminiClient(GeminiConfig())
-            if not client.is_available():
-                return False, [], "GEMINI_API_KEY not set"
-            models = [f"gemini/{m}" for m in client.list_models()]
-            return True, models, None
-        except ValueError as e:
-            return False, [], str(e)
-
-    def get_ollama_models() -> Tuple[bool, List[str], Optional[str]]:
-        """Returns (available, models, error_msg)."""
-        try:
-            client = OllamaClient(OllamaConfig())
-            models = [f"ollama/{m}" for m in client.list_models()]
-            if not models:
-                msg = "No models installed. Run: ollama pull <model>"
-                return True, [], msg
-            return True, models, None
-        except ValueError as e:
-            return False, [], str(e)
-
-    def get_openai_models() -> Tuple[bool, List[str], Optional[str]]:
-        """Returns (available, models, error_msg)."""
-        try:
-            client = OpenAIClient(OpenAIConfig())
-            if not client.is_available():
-                return False, [], "OPENAI_API_KEY not set"
-            models = [f"openai/{m}" for m in client.list_models()]
-            return True, models, None
-        except ValueError as e:
-            return False, [], str(e)
-
-    def get_deepseek_models() -> Tuple[bool, List[str], Optional[str]]:
-        """Returns (available, models, error_msg)."""
-        try:
-            client = DeepSeekClient(DeepSeekConfig())
-            if not client.is_available():
-                return False, [], "DEEPSEEK_API_KEY not set"
-            models = [f"deepseek/{m}" for m in client.list_models()]
-            return True, models, None
-        except ValueError as e:
-            return False, [], str(e)
-
-    def get_mistral_models() -> Tuple[bool, List[str], Optional[str]]:
-        """Returns (available, models, error_msg)."""
-        try:
-            client = MistralClient(MistralConfig())
-            if not client.is_available():
-                return False, [], "MISTRAL_API_KEY not set"
-            models = [f"mistral/{m}" for m in client.list_models()]
-            return True, models, None
-        except ValueError as e:
-            return False, [], str(e)
-
-    providers: List[ProviderInfo] = [
-        {
-            "name": "Groq",
-            "prefix": "groq/",
-            "env_var": "GROQ_API_KEY",
-            "url": "https://console.groq.com",
-            "get_models": get_groq_models,
-        },
-        {
-            "name": "Anthropic",
-            "prefix": "anthropic/",
-            "env_var": "ANTHROPIC_API_KEY",
-            "url": "https://console.anthropic.com",
-            "get_models": get_anthropic_models,
-        },
-        {
-            "name": "Gemini",
-            "prefix": "gemini/",
-            "env_var": "GEMINI_API_KEY",
-            "url": "https://aistudio.google.com",
-            "get_models": get_gemini_models,
-        },
-        {
-            "name": "Ollama (Local)",
-            "prefix": "ollama/",
-            "env_var": None,
-            "url": "https://ollama.ai",
-            "get_models": get_ollama_models,
-        },
-        {
-            "name": "OpenAI",
-            "prefix": "openai/",
-            "env_var": "OPENAI_API_KEY",
-            "url": "https://platform.openai.com",
-            "get_models": get_openai_models,
-        },
-        {
-            "name": "DeepSeek",
-            "prefix": "deepseek/",
-            "env_var": "DEEPSEEK_API_KEY",
-            "url": "https://platform.deepseek.com",
-            "get_models": get_deepseek_models,
-        },
-        {
-            "name": "Mistral",
-            "prefix": "mistral/",
-            "env_var": "MISTRAL_API_KEY",
-            "url": "https://console.mistral.ai",
-            "get_models": get_mistral_models,
-        },
-    ]
-
-    # Filter providers if a specific one is requested
-    valid_provider_names = [
-        "groq",
-        "anthropic",
-        "gemini",
-        "ollama",
-        "openai",
-        "deepseek",
-        "mistral",
-    ]
-    if provider:
-        provider_lower = provider.lower()
-        if provider_lower not in valid_provider_names:
-            click.echo(
-                f"Unknown provider: {provider}. "
-                f"Valid options: {', '.join(valid_provider_names)}",
-                err=True,
-            )
-            sys.exit(1)
-        providers = [
-            p for p in providers if p["prefix"].rstrip("/") == provider_lower
-        ]
-
-    click.echo("\nAvailable LLM Models:\n")
-
-    any_available = False
-    for prov in providers:
-        available, models, error = prov["get_models"]()
-
-        if available and models:
-            any_available = True
-            status = click.style("[OK]", fg="green")
-            count = len(models)
-            click.echo(f"  {status} {prov['name']} ({count} models):")
-            for m in models:
-                click.echo(f"      {m}")
-        elif available and not models:
-            status = click.style("[!]", fg="yellow")
-            click.echo(f"  {status} {prov['name']}:")
-            click.echo(f"      {error}")
-        else:
-            status = click.style("[X]", fg="red")
-            click.echo(f"  {status} {prov['name']}:")
-            click.echo(f"      {error}")
-
-        click.echo()
-
-    click.echo("Provider Setup:")
-    for prov in providers:
-        available, _, _ = prov["get_models"]()
-        if prov["env_var"]:
-            status = "configured" if available else "not set"
-            color = "green" if available else "yellow"
-            click.echo(
-                f"  {prov['env_var']}: "
-                f"{click.style(status, fg=color)} - {prov['url']}"
-            )
-        else:
-            status = "running" if available else "not running"
-            color = "green" if available else "yellow"
-            click.echo(
-                f"  Ollama server: "
-                f"{click.style(status, fg=color)} - {prov['url']}"
-            )
-
-    click.echo()
-    click.echo(
-        click.style("Note: ", fg="yellow")
-        + "Some models may require a paid plan. "
-        + "Free tier availability varies by provider."
-    )
-    click.echo()
-    if any_available:
-        click.echo("Default model: groq/llama-3.1-8b-instant")
-    click.echo()
-
-
-def main() -> None:
-    """Entry point for the CLI."""
-    cli()
-
-
-if __name__ == "__main__":  # pragma: no cover
-    main()

causaliq_knowledge-0.2.0.dist-info/RECORD

@@ -1,22 +0,0 @@
-causaliq_knowledge/__init__.py,sha256=IcoxZ6fjiN6VrniikCUZhHkxf2D1eGixtLWNrvKevN0,851
-causaliq_knowledge/base.py,sha256=GBG-sftOKkmUoQzTpm6anDTjP-2nInRZN_36dxoYhvk,2917
-causaliq_knowledge/cli.py,sha256=2c8WYxF4T_-R8hDIo9JiZCx59fVbXHCCln66UGLqirs,13169
-causaliq_knowledge/models.py,sha256=tWGf186ASwO8NHiN97pEOLuBJmJI6Q9jvpU0mYZNdS0,4058
-causaliq_knowledge/llm/__init__.py,sha256=30AL0h64zIkXoiqhMY7gjaf7mrtwtwMW38vzhns0My4,1663
-causaliq_knowledge/llm/anthropic_client.py,sha256=dPFHYGWL4xwQCtmQuGwGY4DBKSINOgOS-11ekznaiXo,8719
-causaliq_knowledge/llm/base_client.py,sha256=Dg5s9FqtTScliEK9MJ2_B0atTNwRRMNscv9gai6sEB4,7090
-causaliq_knowledge/llm/deepseek_client.py,sha256=ZcOpgnYa66XHjiTaF5ekR_BtosRYvVmzlIafp_Gsx_A,3543
-causaliq_knowledge/llm/gemini_client.py,sha256=XJMq9sPo7zExrALSr2rIRHLheSPqKo8ENG0KtdJ1cjw,9924
-causaliq_knowledge/llm/groq_client.py,sha256=PnTXqtMF1Km9DY4HiCZXQ6LeOzdjZtQJaeuGe1GbeME,7531
-causaliq_knowledge/llm/mistral_client.py,sha256=dTAOtymffCM1AJp5-JcfizofYrUA-jhKfHWrhZe2DDI,4187
-causaliq_knowledge/llm/ollama_client.py,sha256=PPU3g-nD8D546zcYB3uGxZ9yVbU4Gngo3snM2tRFeTc,8612
-causaliq_knowledge/llm/openai_client.py,sha256=MJmB6P32TZESMlXhn9d0-b3vFWXmf7ojHQ5CY8mCENI,3835
-causaliq_knowledge/llm/openai_compat_client.py,sha256=L8ZW5csuhUePq4mt3EGOUqhR3tleFmM72UlhPBsgIMQ,9518
-causaliq_knowledge/llm/prompts.py,sha256=bJ9iVGKUfTfLi2eWh-FFM4cNzk5Ux4Z0x8R6Ia27Dbo,6598
-causaliq_knowledge/llm/provider.py,sha256=VDEv-1esT_EgJk_Gwlfl4423ojglOxzPCBCFbOFE4DQ,15184
-causaliq_knowledge-0.2.0.dist-info/licenses/LICENSE,sha256=vUFUzQnti-D-MLSi9NxFlsFYOKwU25sxxH7WgJOQFIs,1084
-causaliq_knowledge-0.2.0.dist-info/METADATA,sha256=NxnJJjL6hED91fu0DlclGSoeiji8litmEsy1sS_lt_0,8726
-causaliq_knowledge-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-causaliq_knowledge-0.2.0.dist-info/entry_points.txt,sha256=8iQjiMgFxZszRWwSTGHvoOBb_OBUkMmwvH3PzgsH-Cc,104
-causaliq_knowledge-0.2.0.dist-info/top_level.txt,sha256=GcxQf4BQAGa38i2-j8ylk2FmnBHtEZ9-8bSt-7Uka7k,19
-causaliq_knowledge-0.2.0.dist-info/RECORD,,