parseforge 0.2.6__tar.gz → 0.2.9__tar.gz

{parseforge-0.2.6/parseforge.egg-info → parseforge-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: parseforge
-Version: 0.2.6
+Version: 0.2.9
 Summary: LLM-driven pipeline that forges, validates, and promotes TextFSM templates from network CLI output
 Author-email: Tuyen Mathew Duong <tuyen@geekstrident.com>
 Maintainer-email: Tuyen Mathew Duong <tuyen@geekstrident.com>
@@ -60,6 +60,15 @@ Requires-Dist: textfsm-ai[moonshot]>=0.6.1; extra == "moonshot"
 Provides-Extra: cerebras
 Requires-Dist: openai>=1.0.0; extra == "cerebras"
 Requires-Dist: textfsm-ai[cerebras]>=0.6.1; extra == "cerebras"
+Provides-Extra: mistral
+Requires-Dist: mistralai==1.10.0; extra == "mistral"
+Requires-Dist: textfsm-ai[mistral]>=0.6.1; extra == "mistral"
+Provides-Extra: cohere
+Requires-Dist: cohere==5.21.1; extra == "cohere"
+Requires-Dist: textfsm-ai[cohere]>=0.6.1; extra == "cohere"
+Provides-Extra: azure
+Requires-Dist: azure-ai-inference>=1.0.0b9; extra == "azure"
+Requires-Dist: textfsm-ai[azure]>=0.6.1; extra == "azure"
 Provides-Extra: sampling
 Requires-Dist: netmiko>=4.0; extra == "sampling"
 Provides-Extra: dev
@@ -70,6 +79,9 @@ Requires-Dist: black>=24.0.0; extra == "dev"
 Requires-Dist: mypy>=1.10.0; extra == "dev"
 Requires-Dist: anthropic>=0.40.0; extra == "dev"
 Requires-Dist: openai>=1.0.0; extra == "dev"
+Requires-Dist: mistralai==1.10.0; extra == "dev"
+Requires-Dist: cohere==5.21.1; extra == "dev"
+Requires-Dist: azure-ai-inference>=1.0.0b9; extra == "dev"
 Provides-Extra: release
 Requires-Dist: bump2version; extra == "release"
 Requires-Dist: build; extra == "release"
@@ -112,11 +124,11 @@ command that's pure local processing (`canonical`/`readable`/`recognizers`,
 LLM (`name`, `check --provider`, `run`, `generate-template`, `trial`) needs the
 extra for whichever provider it uses: `anthropic`, `openai`, `deepseek`,
 `groq`, `xai`, `together`, `fireworks`, `perplexity`, `openrouter`,
-`moonshot`, or `cerebras`. `--provider` defaults to `anthropic` wherever it
-isn't required, so that's the one most setups need. `pip install
-parseforge[sampling]` adds Netmiko for live device sampling; combine extras
-as needed, e.g.
-`pip install parseforge[anthropic,openai,deepseek,groq,xai,together,fireworks,perplexity,openrouter,moonshot,cerebras,sampling]`.
+`moonshot`, `cerebras`, `mistral`, `cohere`, or `azure`. `--provider` defaults
+to `anthropic` wherever it isn't required, so that's the one most setups
+need. `pip install parseforge[sampling]` adds Netmiko for live device
+sampling; combine extras as needed, e.g.
+`pip install parseforge[anthropic,openai,deepseek,groq,xai,together,fireworks,perplexity,openrouter,moonshot,cerebras,mistral,cohere,azure,sampling]`.
 
 ## Development
 
@@ -124,11 +136,13 @@ as needed, e.g.
 pip install -e ".[dev,sampling]"
 pytest
 ```
-`dev` already includes both the `anthropic` and `openai` SDKs (tests exercise
-all eleven providers — `anthropic`, `openai`, `deepseek`, `groq`, `xai`,
-`together`, `fireworks`, `perplexity`, `openrouter`, `moonshot`, `cerebras`
-share just those two packages — and never silently skip) — add the specific
-`,<provider>` extra explicitly only if installing outside of `dev`.
+`dev` already includes the `anthropic`, `openai`, `mistralai`, `cohere`, and
+`azure-ai-inference` SDKs (tests exercise all fourteen providers —
+`anthropic`, `openai`, `deepseek`, `groq`, `xai`, `together`, `fireworks`,
+`perplexity`, `openrouter`, `moonshot`, `cerebras` share just the first two
+packages, and `mistral`/`cohere`/`azure` each need their own SDK — and never
+silently skip) — add the specific `,<provider>` extra explicitly only if
+installing outside of `dev`.
 
 Linting/formatting/type-checking/docs run through tox instead of extras — see
 `tox.ini` (`tox -e lint`/`format`/`typecheck`/`docs`), each installing its own
@@ -150,8 +164,14 @@ parseforge name --vendor cisco --family catalyst9200 --os ios-xe --version 17.9.
 ```
 Still needs an LLM provider on a cache miss — `--provider` defaults to `anthropic`,
 and `--api-key` falls back to that provider's own env var (`ANTHROPIC_API_KEY`/
-`OPENAI_API_KEY`/`DEEPSEEK_API_KEY`). A cache hit (a command already seen before)
-never touches the LLM, so no key is needed at all in that case.
+`OPENAI_API_KEY`/`DEEPSEEK_API_KEY`/etc.). A cache hit (a command already seen
+before) never touches the LLM, so no key is needed at all in that case.
+
+`--provider azure` is the one exception to the usual `--api-key`/`--model` shape —
+Azure has no fixed base_url or model catalog, so it also needs `--endpoint`
+(`AZURE_ENDPOINT`), `--api-version` (`AZURE_API_VERSION`), and `--deployment`
+(`AZURE_DEPLOYMENT`, replacing `--model`). Same three options exist on
+`check --provider`, `run`, and `generate-template` below.
 
 **`check`** — validate a connector or provider before spending time/tokens on a real
 run. With neither `--env` nor explicit connection flags, prints what a connector needs
@@ -196,7 +216,10 @@ parseforge run --vendor cisco --family catalyst9200 --os ios-xe --version 17.9.1
 `--naming-provider`/`--naming-api-key`/`--naming-model`, defaulting independently
 (`--naming-provider` defaults to `anthropic`; the other two fall back to that
 provider's own env var/default model) — set them explicitly if naming needs a
-different provider than generation.
+different provider than generation. `--provider azure`/`--naming-provider azure`
+each get their own `--endpoint`/`--api-version`/`--deployment` (generation) and
+`--naming-endpoint`/`--naming-api-version`/`--naming-deployment` (naming) — see
+the `name` section above for what they're for.
 
 **`init-trial-config`** — write a placeholder `trial.yaml` to fill in, instead of
 writing one by hand:
@@ -229,7 +252,9 @@ workers: 1
 ```
 `provider`/`api_key`/`model` are one shared LLM source used for both naming and
 generation. Use the `run` command's separate `--naming-*`/`--generation-*` flags
-instead if a trial actually needs two different providers.
+instead if a trial actually needs two different providers. For `provider: azure`,
+also set `endpoint`/`api_version`/`deployment` (`deployment` replaces `model`) —
+see `parseforge init-trial-config`'s generated placeholder for the exact keys.
 
 **`integration`** — rebuild `integration/` for every case under `trials/` (SPEC §5
 step 8):

{parseforge-0.2.6 → parseforge-0.2.9}/README.md

@@ -35,11 +35,11 @@ command that's pure local processing (`canonical`/`readable`/`recognizers`,
 LLM (`name`, `check --provider`, `run`, `generate-template`, `trial`) needs the
 extra for whichever provider it uses: `anthropic`, `openai`, `deepseek`,
 `groq`, `xai`, `together`, `fireworks`, `perplexity`, `openrouter`,
-`moonshot`, or `cerebras`. `--provider` defaults to `anthropic` wherever it
-isn't required, so that's the one most setups need. `pip install
-parseforge[sampling]` adds Netmiko for live device sampling; combine extras
-as needed, e.g.
-`pip install parseforge[anthropic,openai,deepseek,groq,xai,together,fireworks,perplexity,openrouter,moonshot,cerebras,sampling]`.
+`moonshot`, `cerebras`, `mistral`, `cohere`, or `azure`. `--provider` defaults
+to `anthropic` wherever it isn't required, so that's the one most setups
+need. `pip install parseforge[sampling]` adds Netmiko for live device
+sampling; combine extras as needed, e.g.
+`pip install parseforge[anthropic,openai,deepseek,groq,xai,together,fireworks,perplexity,openrouter,moonshot,cerebras,mistral,cohere,azure,sampling]`.
 
 ## Development
 
@@ -47,11 +47,13 @@ as needed, e.g.
 pip install -e ".[dev,sampling]"
 pytest
 ```
-`dev` already includes both the `anthropic` and `openai` SDKs (tests exercise
-all eleven providers — `anthropic`, `openai`, `deepseek`, `groq`, `xai`,
-`together`, `fireworks`, `perplexity`, `openrouter`, `moonshot`, `cerebras`
-share just those two packages — and never silently skip) — add the specific
-`,<provider>` extra explicitly only if installing outside of `dev`.
+`dev` already includes the `anthropic`, `openai`, `mistralai`, `cohere`, and
+`azure-ai-inference` SDKs (tests exercise all fourteen providers —
+`anthropic`, `openai`, `deepseek`, `groq`, `xai`, `together`, `fireworks`,
+`perplexity`, `openrouter`, `moonshot`, `cerebras` share just the first two
+packages, and `mistral`/`cohere`/`azure` each need their own SDK — and never
+silently skip) — add the specific `,<provider>` extra explicitly only if
+installing outside of `dev`.
 
 Linting/formatting/type-checking/docs run through tox instead of extras — see
 `tox.ini` (`tox -e lint`/`format`/`typecheck`/`docs`), each installing its own
@@ -73,8 +75,14 @@ parseforge name --vendor cisco --family catalyst9200 --os ios-xe --version 17.9.
 ```
 Still needs an LLM provider on a cache miss — `--provider` defaults to `anthropic`,
 and `--api-key` falls back to that provider's own env var (`ANTHROPIC_API_KEY`/
-`OPENAI_API_KEY`/`DEEPSEEK_API_KEY`). A cache hit (a command already seen before)
-never touches the LLM, so no key is needed at all in that case.
+`OPENAI_API_KEY`/`DEEPSEEK_API_KEY`/etc.). A cache hit (a command already seen
+before) never touches the LLM, so no key is needed at all in that case.
+
+`--provider azure` is the one exception to the usual `--api-key`/`--model` shape —
+Azure has no fixed base_url or model catalog, so it also needs `--endpoint`
+(`AZURE_ENDPOINT`), `--api-version` (`AZURE_API_VERSION`), and `--deployment`
+(`AZURE_DEPLOYMENT`, replacing `--model`). Same three options exist on
+`check --provider`, `run`, and `generate-template` below.
 
 **`check`** — validate a connector or provider before spending time/tokens on a real
 run. With neither `--env` nor explicit connection flags, prints what a connector needs
@@ -119,7 +127,10 @@ parseforge run --vendor cisco --family catalyst9200 --os ios-xe --version 17.9.1
 `--naming-provider`/`--naming-api-key`/`--naming-model`, defaulting independently
 (`--naming-provider` defaults to `anthropic`; the other two fall back to that
 provider's own env var/default model) — set them explicitly if naming needs a
-different provider than generation.
+different provider than generation. `--provider azure`/`--naming-provider azure`
+each get their own `--endpoint`/`--api-version`/`--deployment` (generation) and
+`--naming-endpoint`/`--naming-api-version`/`--naming-deployment` (naming) — see
+the `name` section above for what they're for.
 
 **`init-trial-config`** — write a placeholder `trial.yaml` to fill in, instead of
 writing one by hand:
@@ -152,7 +163,9 @@ workers: 1
 ```
 `provider`/`api_key`/`model` are one shared LLM source used for both naming and
 generation. Use the `run` command's separate `--naming-*`/`--generation-*` flags
-instead if a trial actually needs two different providers.
+instead if a trial actually needs two different providers. For `provider: azure`,
+also set `endpoint`/`api_version`/`deployment` (`deployment` replaces `model`) —
+see `parseforge init-trial-config`'s generated placeholder for the exact keys.
 
 **`integration`** — rebuild `integration/` for every case under `trials/` (SPEC §5
 step 8):

parseforge-0.2.9/parseforge/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.2.9"

{parseforge-0.2.6 → parseforge-0.2.9}/parseforge/cli/config.py

@@ -42,6 +42,14 @@ class TrialConfig:
     note: str | None = None
     workers: int = 1
     path: str | None = None
+    # Only meaningful for provider: azure — no fixed base_url or model
+    # catalog there (see naming/providers/azure.py). endpoint/api_version
+    # are ignored by every other provider; deployment replaces the usual
+    # "model" concept and is what actually gets sent as the generation
+    # model argument (see cli/main.py's trial_cmd).
+    endpoint: str | None = None
+    api_version: str | None = None
+    deployment: str | None = None
 
 
 @dataclass(frozen=True)
@@ -57,6 +65,10 @@ class GenerationRequestConfig:
     env: str | None = None
     cmdline: str | None = None
     sample_file: str | None = None
+    # Only meaningful for provider: azure — see TrialConfig above.
+    endpoint: str | None = None
+    api_version: str | None = None
+    deployment: str | None = None
 
 
 _TRIAL_REQUIRED = (
@@ -124,6 +136,9 @@ def load_trial_config(path: Path) -> TrialConfig:
         note=raw.get("notes"),
         workers=int(raw.get("workers", 1)),
         path=raw.get("path"),
+        endpoint=raw.get("endpoint"),
+        api_version=raw.get("api_version"),
+        deployment=raw.get("deployment"),
     )
 
 
@@ -143,4 +158,7 @@ def load_generation_config(path: Path) -> GenerationRequestConfig:
         env=raw.get("env"),
         cmdline=raw.get("cmdline"),
         sample_file=raw.get("sample_file"),
+        endpoint=raw.get("endpoint"),
+        api_version=raw.get("api_version"),
+        deployment=raw.get("deployment"),
     )

{parseforge-0.2.6 → parseforge-0.2.9}/parseforge/cli/main.py

@@ -20,13 +20,17 @@ from parseforge import (
     validation,
 )
 from parseforge.cli import config as cli_config
+from parseforge.naming.providers.azure import DEFAULT_API_VERSION
 
 _BUILDERS: dict[str, type[naming.RegexBuilder]] = {
     "anthropic": naming.AnthropicRegexBuilder,
+    "azure": naming.AzureRegexBuilder,
     "cerebras": naming.CerebrasRegexBuilder,
+    "cohere": naming.CohereRegexBuilder,
     "deepseek": naming.DeepSeekRegexBuilder,
     "fireworks": naming.FireworksRegexBuilder,
     "groq": naming.GroqRegexBuilder,
+    "mistral": naming.MistralRegexBuilder,
     "moonshot": naming.MoonshotRegexBuilder,
     "openai": naming.OpenAIRegexBuilder,
     "openrouter": naming.OpenRouterRegexBuilder,
@@ -59,6 +63,13 @@ device_type: <netmiko-device-type>
 provider: anthropic
 api_key: <api-key>
 model: <model>
+# For provider: azure only -- no fixed base_url or model catalog there.
+# deployment replaces model above (still keep a model: placeholder value,
+# it's simply ignored); endpoint/api_version default to AZURE_ENDPOINT/
+# AZURE_API_VERSION if omitted.
+# endpoint: <azure-resource-endpoint>
+# api_version: <azure-api-version>
+# deployment: <azure-deployment-name>
 
 # --- commands to run (required, at least one) ---
 commands:
@@ -84,6 +95,10 @@ _GENERATE_TEMPLATE_CONFIG_PLACEHOLDER = """\
 provider: anthropic
 api_key: <api-key>
 model: <model>
+# For provider: azure only -- see trial.yaml's placeholder for details.
+# endpoint: <azure-resource-endpoint>
+# api_version: <azure-api-version>
+# deployment: <azure-deployment-name>
 
 # --- input: choose exactly one ---
 
@@ -112,13 +127,34 @@ def _write_placeholder_config(out_path: str, force: bool, content: str) -> None:
 
 
 def _build_regex_builder(
-    provider: str, api_key: str | None, model: str | None
+    provider: str,
+    api_key: str | None,
+    model: str | None,
+    endpoint: str | None = None,
+    api_version: str | None = None,
+    deployment: str | None = None,
 ) -> naming.RegexBuilder:
+    """endpoint/api_version/deployment are only meaningful for
+    provider="azure" (no fixed base_url or model catalog there — see
+    naming/providers/azure.py) — forwarded only when given, so every
+    other provider's constructor never sees an unexpected kwarg.
+
+    When deployment is given, model is *not* forwarded — AzureRegexBuilder
+    has no model parameter at all (deployment replaces it), so passing
+    both would raise a plain TypeError. This matters in practice: every
+    TrialConfig always has a (placeholder) model value, since it's a
+    required config key shared by every provider."""
     kwargs: dict[str, str] = {}
     if api_key is not None:
         kwargs["api_key"] = api_key
-    if model is not None:
+    if deployment is not None:
+        kwargs["deployment"] = deployment
+    elif model is not None:
         kwargs["model"] = model
+    if endpoint is not None:
+        kwargs["endpoint"] = endpoint
+    if api_version is not None:
+        kwargs["api_version"] = api_version
     return _BUILDERS[provider](**kwargs)
 
 
@@ -209,13 +245,32 @@ def main() -> None:
     help="Provider API key. Defaults to that provider's own API key environment "
     "variable (ANTHROPIC_API_KEY, OPENAI_API_KEY, DEEPSEEK_API_KEY, GROQ_API_KEY, "
     "XAI_API_KEY, TOGETHER_API_KEY, FIREWORKS_API_KEY, PERPLEXITY_API_KEY, "
-    "OPENROUTER_API_KEY, MOONSHOT_API_KEY, CEREBRAS_API_KEY); only needed on a "
-    "cache miss.",
+    "OPENROUTER_API_KEY, MOONSHOT_API_KEY, CEREBRAS_API_KEY, MISTRAL_API_KEY, "
+    "COHERE_API_KEY, AZURE_API_KEY); only needed on a cache miss.",
 )
 @click.option(
     "--model",
     default=None,
-    help="Defaults to the selected provider's own default model.",
+    help="Defaults to the selected provider's own default model. Ignored for "
+    "--provider azure — use --deployment instead.",
+)
+@click.option(
+    "--endpoint",
+    default=None,
+    help="Azure resource endpoint. Only used by --provider azure; defaults to "
+    "AZURE_ENDPOINT.",
+)
+@click.option(
+    "--api-version",
+    default=None,
+    help="Azure API version. Only used by --provider azure; defaults to "
+    "AZURE_API_VERSION, then a built-in default.",
+)
+@click.option(
+    "--deployment",
+    default=None,
+    help="Azure deployment name — replaces --model for --provider azure; "
+    "defaults to AZURE_DEPLOYMENT.",
 )
 @click.argument("command", nargs=-1, required=True)
 def name_cmd(
@@ -226,6 +281,9 @@ def name_cmd(
     provider: str,
     api_key: str | None,
     model: str | None,
+    endpoint: str | None,
+    api_version: str | None,
+    deployment: str | None,
     command: tuple[str, ...],
 ) -> None:
     """Print the canonical cli-name for a raw CLI COMMAND.
@@ -234,7 +292,9 @@ def name_cmd(
     on a cache miss.
     """
     context = naming.CliContext(vendor=vendor, family=family, os=os_, version=version)
-    builder = _build_regex_builder(provider, api_key, model)
+    builder = _build_regex_builder(
+        provider, api_key, model, endpoint, api_version, deployment
+    )
     click.echo(naming.cli_name(" ".join(command), context, builder=builder))
 
 
@@ -272,18 +332,66 @@ def name_cmd(
 @click.option(
     "--naming-model",
     default=None,
-    help="Defaults to the naming provider's own default model.",
+    help="Defaults to the naming provider's own default model. Ignored for "
+    "--naming-provider azure — use --naming-deployment instead.",
+)
+@click.option(
+    "--naming-endpoint",
+    default=None,
+    help="Azure resource endpoint for naming. Only used by --naming-provider "
+    "azure; defaults to AZURE_ENDPOINT.",
+)
+@click.option(
+    "--naming-api-version",
+    default=None,
+    help="Azure API version for naming. Only used by --naming-provider azure; "
+    "defaults to AZURE_API_VERSION, then a built-in default.",
+)
+@click.option(
+    "--naming-deployment",
+    default=None,
+    help="Azure deployment name for naming — replaces --naming-model for "
+    "--naming-provider azure; defaults to AZURE_DEPLOYMENT.",
 )
 @click.option(
     "--provider",
     required=True,
     help="LLM provider for template generation (textfsm-ai's own registry, "
     'e.g. "anthropic", "openai", "deepseek", "groq", "xai", "together", '
-    '"fireworks", "perplexity", "openrouter", "moonshot", "cerebras"). Naming '
-    "uses its own separate --naming-provider, not this one.",
+    '"fireworks", "perplexity", "openrouter", "moonshot", "cerebras", '
+    '"mistral", "cohere", "azure"). Naming uses its own separate '
+    "--naming-provider, not this one.",
 )
 @click.option("--api-key", required=True, help="API key for the generation LLM call.")
-@click.option("--model", required=True, help="Model for the generation LLM call.")
+@click.option(
+    "--model",
+    required=True,
+    help="Model for the generation LLM call. For --provider azure, pass the "
+    "deployment name here via --deployment instead — this becomes the "
+    "deployment name that's actually sent.",
+)
+@click.option(
+    "--endpoint",
+    default=None,
+    envvar="AZURE_ENDPOINT",
+    help="Azure resource endpoint for generation. Only used by --provider "
+    "azure; defaults to the AZURE_ENDPOINT environment variable.",
+)
+@click.option(
+    "--api-version",
+    default=None,
+    envvar="AZURE_API_VERSION",
+    help="Azure API version for generation. Only used by --provider azure; "
+    "defaults to the AZURE_API_VERSION environment variable.",
+)
+@click.option(
+    "--deployment",
+    default=None,
+    envvar="AZURE_DEPLOYMENT",
+    help="Azure deployment name for generation — overrides --model for "
+    "--provider azure; defaults to the AZURE_DEPLOYMENT environment "
+    "variable.",
+)
 @click.option(
     "--store-root",
     default=None,
@@ -305,9 +413,15 @@ def run_cmd(
     naming_provider: str,
     naming_api_key: str | None,
     naming_model: str | None,
+    naming_endpoint: str | None,
+    naming_api_version: str | None,
+    naming_deployment: str | None,
     provider: str,
     api_key: str,
     model: str,
+    endpoint: str | None,
+    api_version: str | None,
+    deployment: str | None,
     store_root: str | None,
     project: str | None,
     email: str | None,
@@ -332,9 +446,20 @@ def run_cmd(
     connection = DeviceConnection(
         host=host, username=username, password=password, device_type=device_type
     )
-    naming_builder = _build_regex_builder(naming_provider, naming_api_key, naming_model)
+    naming_builder = _build_regex_builder(
+        naming_provider,
+        naming_api_key,
+        naming_model,
+        naming_endpoint,
+        naming_api_version,
+        naming_deployment,
+    )
     generation_config = LLMProviderConfig(
-        provider=provider, api_key=api_key, model=model
+        provider=provider,
+        api_key=api_key,
+        model=deployment or model,
+        endpoint=endpoint,
+        api_version=api_version,
     )
     metadata = TrialMetadata(
         project=project,
@@ -388,8 +513,17 @@ def _check_connector(
     click.echo("OK")
 
 
-def _check_provider(provider: str, api_key: str | None, model: str | None) -> None:
-    builder = _build_regex_builder(provider, api_key, model)
+def _check_provider(
+    provider: str,
+    api_key: str | None,
+    model: str | None,
+    endpoint: str | None = None,
+    api_version: str | None = None,
+    deployment: str | None = None,
+) -> None:
+    builder = _build_regex_builder(
+        provider, api_key, model, endpoint, api_version, deployment
+    )
     context = naming.CliContext(vendor="check", family="check", os="check", version="0")
     try:
         response = builder.build_pattern("show clock", context)
@@ -422,6 +556,21 @@ def _check_provider(provider: str, api_key: str | None, model: str | None) -> No
 )
 @click.option("--api-key", default=None, help="Used for a --provider check.")
 @click.option("--model", default=None, help="Used for a --provider check.")
+@click.option(
+    "--endpoint",
+    default=None,
+    help="Used for a --provider azure check; defaults to AZURE_ENDPOINT.",
+)
+@click.option(
+    "--api-version",
+    default=None,
+    help="Used for a --provider azure check; defaults to AZURE_API_VERSION.",
+)
+@click.option(
+    "--deployment",
+    default=None,
+    help="Used for a --provider azure check; defaults to AZURE_DEPLOYMENT.",
+)
 def check_cmd(
     connector: str | None,
     provider: str | None,
@@ -433,6 +582,9 @@ def check_cmd(
     probe_command: str,
     api_key: str | None,
     model: str | None,
+    endpoint: str | None,
+    api_version: str | None,
+    deployment: str | None,
 ) -> None:
     """Validate a --connector or --provider is reachable, or report what it
     needs. Exactly one of --connector/--provider is required."""
@@ -444,7 +596,7 @@ def check_cmd(
         )
     else:
         assert provider is not None
-        _check_provider(provider, api_key, model)
+        _check_provider(provider, api_key, model, endpoint, api_version, deployment)
 
 
 @main.command("init-generate-template-config")
@@ -489,7 +641,34 @@ def init_generate_template_config_cmd(out_path: str, force: bool) -> None:
     help='Generation LLM provider (textfsm-ai\'s own registry), e.g. "anthropic".',
 )
 @click.option("--api-key", default=None)
-@click.option("--model", default=None)
+@click.option(
+    "--model",
+    default=None,
+    help="Model for the generation LLM call. For --provider azure, pass the "
+    "deployment name via --deployment instead — this becomes the "
+    "deployment name that's actually sent.",
+)
+@click.option(
+    "--endpoint",
+    default=None,
+    envvar="AZURE_ENDPOINT",
+    help="Azure resource endpoint. Only used by --provider azure; defaults to "
+    "the AZURE_ENDPOINT environment variable.",
+)
+@click.option(
+    "--api-version",
+    default=None,
+    envvar="AZURE_API_VERSION",
+    help="Azure API version. Only used by --provider azure; defaults to the "
+    "AZURE_API_VERSION environment variable.",
+)
+@click.option(
+    "--deployment",
+    default=None,
+    envvar="AZURE_DEPLOYMENT",
+    help="Azure deployment name — overrides --model for --provider azure; "
+    "defaults to the AZURE_DEPLOYMENT environment variable.",
+)
 @click.option(
     "--out",
     "out_dir",
@@ -510,6 +689,9 @@ def generate_template_cmd(
     provider: str | None,
     api_key: str | None,
     model: str | None,
+    endpoint: str | None,
+    api_version: str | None,
+    deployment: str | None,
     out_dir: str | None,
 ) -> None:
     """One-shot template generation — no trial persisted under trials/.
@@ -530,6 +712,9 @@ def generate_template_cmd(
         provider = provider or cfg.provider
         api_key = api_key or cfg.api_key
         model = model or cfg.model
+        endpoint = endpoint or cfg.endpoint
+        api_version = api_version or cfg.api_version
+        deployment = deployment or cfg.deployment
 
     if not provider or not model or not api_key:
         raise click.UsageError(
@@ -549,7 +734,14 @@ def generate_template_cmd(
             "or --host/--username/--password/--device-type), or --config"
         )
 
-    result = generation.generate(sample_text, provider, api_key, model)
+    result = generation.generate(
+        sample_text,
+        provider,
+        api_key,
+        deployment or model,
+        endpoint=endpoint or "",
+        api_version=api_version or DEFAULT_API_VERSION,
+    )
     if not result.ready:
         raise click.ClickException(f"generation not ready: {result.reason}")
 
@@ -682,9 +874,20 @@ def trial_cmd(config_path: str, store_root_opt: str | None) -> None:
         password=cfg.password,
         device_type=cfg.device_type,
     )
-    naming_builder = _build_regex_builder(cfg.provider, cfg.api_key, cfg.model)
+    naming_builder = _build_regex_builder(
+        cfg.provider,
+        cfg.api_key,
+        cfg.model,
+        cfg.endpoint,
+        cfg.api_version,
+        cfg.deployment,
+    )
     generation_config = pipeline.LLMProviderConfig(
-        provider=cfg.provider, api_key=cfg.api_key, model=cfg.model
+        provider=cfg.provider,
+        api_key=cfg.api_key,
+        model=cfg.deployment or cfg.model,
+        endpoint=cfg.endpoint,
+        api_version=cfg.api_version,
     )
     metadata = pipeline.TrialMetadata(
         username=cfg.user, email=cfg.email, description=cfg.description, note=cfg.note

{parseforge-0.2.6 → parseforge-0.2.9}/parseforge/naming/__init__.py

@@ -10,10 +10,13 @@ from .llm import (
 )
 from .providers import (
     AnthropicRegexBuilder,
+    AzureRegexBuilder,
     CerebrasRegexBuilder,
+    CohereRegexBuilder,
     DeepSeekRegexBuilder,
     FireworksRegexBuilder,
     GroqRegexBuilder,
+    MistralRegexBuilder,
     MoonshotRegexBuilder,
     OpenAIRegexBuilder,
     OpenRouterRegexBuilder,
@@ -31,10 +34,13 @@ __all__ = [
     "RegexBuilder",
     "UnimplementedRegexBuilder",
     "AnthropicRegexBuilder",
+    "AzureRegexBuilder",
     "CerebrasRegexBuilder",
+    "CohereRegexBuilder",
     "DeepSeekRegexBuilder",
     "FireworksRegexBuilder",
     "GroqRegexBuilder",
+    "MistralRegexBuilder",
     "MoonshotRegexBuilder",
     "OpenAIRegexBuilder",
     "OpenRouterRegexBuilder",