parseforge 0.2.8__tar.gz → 0.2.9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: parseforge
-Version: 0.2.8
+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>
@@ -66,6 +66,9 @@ 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
@@ -78,6 +81,7 @@ 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"
@@ -120,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`, `cerebras`, `mistral`, or `cohere`. `--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,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
 
@@ -132,13 +136,13 @@ combine extras as needed, e.g.
 pip install -e ".[dev,sampling]"
 pytest
 ```
-`dev` already includes the `anthropic`, `openai`, `mistralai`, and `cohere`
-SDKs (tests exercise all thirteen providers — `anthropic`, `openai`,
-`deepseek`, `groq`, `xai`, `together`, `fireworks`, `perplexity`,
-`openrouter`, `moonshot`, `cerebras` share just the first two packages, and
-`mistral`/`cohere` each need their own native SDK — 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
@@ -160,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
@@ -206,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:
@@ -239,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.8 → 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`, `cerebras`, `mistral`, or `cohere`. `--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,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,13 +47,13 @@ combine extras as needed, e.g.
 pip install -e ".[dev,sampling]"
 pytest
 ```
-`dev` already includes the `anthropic`, `openai`, `mistralai`, and `cohere`
-SDKs (tests exercise all thirteen providers — `anthropic`, `openai`,
-`deepseek`, `groq`, `xai`, `together`, `fireworks`, `perplexity`,
-`openrouter`, `moonshot`, `cerebras` share just the first two packages, and
-`mistral`/`cohere` each need their own native SDK — 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
@@ -75,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
@@ -121,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:
@@ -154,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.8 → 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.8 → parseforge-0.2.9}/parseforge/cli/main.py

@@ -20,9 +20,11 @@ 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,
@@ -61,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:
@@ -86,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 ---
 
@@ -114,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)
 
 
@@ -212,12 +246,31 @@ def main() -> None:
     "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, MISTRAL_API_KEY, "
-    "COHERE_API_KEY); only needed on a cache miss.",
+    "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(
@@ -228,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.
@@ -236,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))
 
 
@@ -274,7 +332,26 @@ 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",
@@ -282,11 +359,39 @@ def name_cmd(
     help="LLM provider for template generation (textfsm-ai's own registry, "
     'e.g. "anthropic", "openai", "deepseek", "groq", "xai", "together", '
     '"fireworks", "perplexity", "openrouter", "moonshot", "cerebras", '
-    '"mistral", "cohere"). Naming uses its own separate --naming-provider, '
-    "not this one.",
+    '"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,
@@ -308,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,
@@ -335,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,
@@ -391,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)
@@ -425,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,
@@ -436,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."""
@@ -447,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")
@@ -492,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",
@@ -513,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/.
@@ -533,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(
@@ -552,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}")
 
@@ -685,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.8 → parseforge-0.2.9}/parseforge/naming/__init__.py

@@ -10,6 +10,7 @@ from .llm import (
 )
 from .providers import (
     AnthropicRegexBuilder,
+    AzureRegexBuilder,
     CerebrasRegexBuilder,
     CohereRegexBuilder,
     DeepSeekRegexBuilder,
@@ -33,6 +34,7 @@ __all__ = [
     "RegexBuilder",
     "UnimplementedRegexBuilder",
     "AnthropicRegexBuilder",
+    "AzureRegexBuilder",
     "CerebrasRegexBuilder",
     "CohereRegexBuilder",
     "DeepSeekRegexBuilder",

{parseforge-0.2.8 → parseforge-0.2.9}/parseforge/naming/providers/__init__.py

@@ -1,4 +1,5 @@
 from .anthropic import AnthropicRegexBuilder
+from .azure import AzureRegexBuilder
 from .cerebras import CerebrasRegexBuilder
 from .cohere import CohereRegexBuilder
 from .deepseek import DeepSeekRegexBuilder
@@ -14,6 +15,7 @@ from .xai import XAIRegexBuilder
 
 __all__ = [
     "AnthropicRegexBuilder",
+    "AzureRegexBuilder",
     "CerebrasRegexBuilder",
     "CohereRegexBuilder",
     "DeepSeekRegexBuilder",

parseforge-0.2.9/parseforge/naming/providers/azure.py

@@ -0,0 +1,197 @@
+"""Azure OpenAI-backed RegexBuilder implementation.
+
+Unlike every other provider, Azure has no fixed base_url or model catalog:
+the endpoint is the caller's own Azure resource, and the usual "model"
+concept is replaced by an account-specific deployment name. Talks to the
+``azure-ai-inference`` SDK's ``ChatCompletionsClient`` directly — not
+OpenAI-compatible, not Anthropic-style, not Mistral/Cohere-style either,
+though its response shape (``choices[0].message.content``,
+``finish_reason``, ``usage.prompt_tokens/completion_tokens/total_tokens``)
+happens to closely mirror OpenAI's.
+"""
+
+from __future__ import annotations
+
+import os
+import time
+from typing import TYPE_CHECKING, Any
+
+from ..llm import CliContext, LLMCLIResponse, TokenUsage, build_prompt
+from .cost import estimate_cost
+from .text import extract_pattern
+
+if TYPE_CHECKING:
+    from azure.ai.inference import ChatCompletionsClient as _ChatCompletionsClient
+
+_DEFAULT_MAX_TOKENS = 1024
+
+# Matches textfsm-ai's own AzureOpenAIProvider.from_env() default.
+DEFAULT_API_VERSION = "2024-02-15-preview"
+
+# HTTP statuses where retrying the exact same request can't succeed —
+# mirrors providers/errors.py's NON_RETRYABLE class-name set, but the
+# azure-ai-inference SDK raises a single HttpResponseError for every HTTP
+# failure (status on exc.status_code, populated from the underlying HTTP
+# response) rather than a family of named exception classes, so
+# classification here is done by status code — same approach as Mistral/
+# Cohere.
+_NON_RETRYABLE_STATUSES = frozenset({400, 401, 403, 404, 409, 413, 422})
+
+
+def _import_azure() -> Any:
+    """Deferred import — azure-ai-inference is an optional extra
+    (parseforge[azure]); nothing else in this module (or a cache-hit
+    lookup, which never reaches build_pattern at all — see
+    resolver.cli_name) should require it to be installed."""
+    try:
+        from azure.ai.inference import ChatCompletionsClient
+        from azure.core.credentials import AzureKeyCredential
+        from azure.core.exceptions import HttpResponseError
+    except ImportError as exc:
+        raise ImportError(
+            "the azure-ai-inference package is required to use "
+            "AzureRegexBuilder — install it via `pip install parseforge[azure]`"
+        ) from exc
+    return ChatCompletionsClient, AzureKeyCredential, HttpResponseError
+
+
+def _build_deployment_endpoint(endpoint: str, deployment: str) -> str:
+    """Normalize a base Azure resource endpoint into the full deployment
+    endpoint azure-ai-inference's ChatCompletionsClient expects — mirrors
+    textfsm-ai's own build_azure_endpoint(). If the caller already passed
+    a full deployment endpoint, it's returned unchanged."""
+    endpoint = endpoint.rstrip("/")
+    if "/openai/deployments/" in endpoint:
+        return endpoint
+    return f"{endpoint}/openai/deployments/{deployment}"
+
+
+def _is_retryable(exc: Any) -> bool:
+    return getattr(exc, "status_code", None) not in _NON_RETRYABLE_STATUSES
+
+
+def _format_error_reason(exc: Any) -> str:
+    return f"LLM-ERROR-azure_sdk-{getattr(exc, 'status_code', None)}-{exc}"
+
+
+class AzureRegexBuilder:
+    """Builds a cli-name regex pattern by prompting an Azure OpenAI deployment.
+
+    Unlike every other provider, there's no fixed model catalog — the
+    ``deployment`` argument (an account-specific deployment name) is
+    required, resolving from the argument if given, otherwise the
+    ``AZURE_DEPLOYMENT`` environment variable. ``endpoint`` (the caller's
+    own Azure resource URL) similarly resolves from the argument, then
+    ``AZURE_ENDPOINT``. ``api_version`` resolves from the argument, then
+    ``AZURE_API_VERSION``, then a default matching textfsm-ai's own
+    AzureOpenAIProvider.
+
+    The client is constructed lazily, on the first actual call — not in
+    ``__init__`` — so this can be used as a default RegexBuilder without
+    requiring any of these to be set for cache-hit lookups, which never
+    reach the LLM at all (see resolver.cli_name).
+    """
+
+    provider = "azure"
+
+    def __init__(
+        self,
+        deployment: str | None = None,
+        api_key: str | None = None,
+        endpoint: str | None = None,
+        api_version: str | None = None,
+    ) -> None:
+        self._deployment = deployment
+        self._api_key = api_key
+        self._endpoint = endpoint
+        self._api_version = api_version
+        self._resolved_deployment: str | None = None
+        self._client: _ChatCompletionsClient | None = None
+
+    def _get_client(self) -> _ChatCompletionsClient:
+        if self._client is None:
+            ChatCompletionsClient, AzureKeyCredential, _ = _import_azure()
+            api_key = self._api_key or os.environ.get("AZURE_API_KEY")
+            if not api_key:
+                raise RuntimeError(
+                    "no Azure API key — pass api_key or set AZURE_API_KEY"
+                )
+            endpoint = self._endpoint or os.environ.get("AZURE_ENDPOINT")
+            if not endpoint:
+                raise RuntimeError(
+                    "no Azure endpoint — pass endpoint or set AZURE_ENDPOINT"
+                )
+            deployment = self._deployment or os.environ.get("AZURE_DEPLOYMENT")
+            if not deployment:
+                raise RuntimeError(
+                    "no Azure deployment — pass deployment or set AZURE_DEPLOYMENT"
+                )
+            api_version = (
+                self._api_version
+                or os.environ.get("AZURE_API_VERSION")
+                or DEFAULT_API_VERSION
+            )
+            self._resolved_deployment = deployment
+            self._client = ChatCompletionsClient(
+                endpoint=_build_deployment_endpoint(endpoint, deployment),
+                credential=AzureKeyCredential(api_key),
+                api_version=api_version,
+            )
+        return self._client
+
+    def build_pattern(
+        self, command: str, context: CliContext, **kwargs: Any
+    ) -> LLMCLIResponse:
+        _, _, HttpResponseError = _import_azure()
+        prompt = build_prompt(command, context)
+        max_tokens = kwargs.pop("max_tokens", None) or _DEFAULT_MAX_TOKENS
+
+        start = time.monotonic()
+        try:
+            response = self._get_client().complete(
+                messages=[{"role": "user", "content": prompt}],
+                max_tokens=max_tokens,
+                **kwargs,
+            )
+        except HttpResponseError as exc:
+            if not _is_retryable(exc):
+                # Same request would fail the same way again — stop rather
+                # than let a caller burn another attempt on it.
+                raise
+            return LLMCLIResponse(
+                content="",
+                raw=exc,
+                usage=TokenUsage(
+                    input_tokens=0, output_tokens=0, total_tokens=0, estimated_cost=0.0
+                ),
+                duration_ms=(time.monotonic() - start) * 1000,
+                reason=_format_error_reason(exc),
+                ready=False,
+            )
+        duration_ms = (time.monotonic() - start) * 1000
+
+        choice = response.choices[0]
+        return LLMCLIResponse(
+            content=extract_pattern(choice.message.content or ""),
+            raw=response,
+            usage=TokenUsage(
+                input_tokens=response.usage.prompt_tokens,
+                output_tokens=response.usage.completion_tokens,
+                total_tokens=response.usage.total_tokens,
+                estimated_cost=estimate_cost(
+                    input_tokens=response.usage.prompt_tokens,
+                    output_tokens=response.usage.completion_tokens,
+                    total_tokens=response.usage.total_tokens,
+                    provider=self.provider,
+                    # No public pricing table for a customer's own Azure
+                    # deployment — estimate_cost() falls back to 0.0 with
+                    # a warning for an unrecognized (provider, model) pair,
+                    # which is the correct/expected outcome here, not an
+                    # error condition.
+                    model=self._resolved_deployment or "",
+                ),
+            ),
+            duration_ms=duration_ms,
+            reason=choice.finish_reason or "",
+            ready=choice.finish_reason == "stop",
+        )

{parseforge-0.2.8 → parseforge-0.2.9}/parseforge/pipeline.py

@@ -46,6 +46,7 @@ from pathlib import Path
 from typing import Any
 
 from parseforge import generation, naming, paths, sampling, validation
+from parseforge.naming.providers.azure import DEFAULT_API_VERSION
 
 
 class Mode(str, Enum):
@@ -55,9 +56,18 @@ class Mode(str, Enum):
 
 @dataclass(frozen=True)
 class LLMProviderConfig:
+    """``model`` doubles as the deployment name for provider="azure" (no
+    fixed model catalog there — see naming/providers/azure.py); the CLI
+    resolves --deployment into this field so this dataclass and
+    generation.generate()'s positional model arg stay untouched.
+    ``endpoint``/``api_version`` are ignored by every provider except
+    azure (see textfsm-ai's generation_engine.run())."""
+
     provider: str
     api_key: str
     model: str
+    endpoint: str | None = None
+    api_version: str | None = None
 
 
 @dataclass(frozen=True)
@@ -153,6 +163,13 @@ def run_command_pipeline(
         generation_config.provider,
         generation_config.api_key,
         generation_config.model,
+        endpoint=generation_config.endpoint or "",
+        # Only meaningful for provider="azure" (ignored by every other
+        # provider — see textfsm-ai's generation_engine.run()), which has
+        # no from_env()-style fallback of its own on this path, unlike
+        # naming/providers/azure.py's own AzureRegexBuilder — so the same
+        # default is applied here explicitly.
+        api_version=generation_config.api_version or DEFAULT_API_VERSION,
     )
     (derive_dir / "llm-template.textfsm").write_text(
         gen_result.raw_template, encoding="utf-8"

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: parseforge
-Version: 0.2.8
+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>
@@ -66,6 +66,9 @@ 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
@@ -78,6 +81,7 @@ 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"
@@ -120,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`, `cerebras`, `mistral`, or `cohere`. `--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,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
 
@@ -132,13 +136,13 @@ combine extras as needed, e.g.
 pip install -e ".[dev,sampling]"
 pytest
 ```
-`dev` already includes the `anthropic`, `openai`, `mistralai`, and `cohere`
-SDKs (tests exercise all thirteen providers — `anthropic`, `openai`,
-`deepseek`, `groq`, `xai`, `together`, `fireworks`, `perplexity`,
-`openrouter`, `moonshot`, `cerebras` share just the first two packages, and
-`mistral`/`cohere` each need their own native SDK — 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
@@ -160,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
@@ -206,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:
@@ -239,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.8 → parseforge-0.2.9}/parseforge.egg-info/SOURCES.txt

@@ -29,6 +29,7 @@ parseforge/naming/prompts.yaml
 parseforge/naming/resolver.py
 parseforge/naming/providers/__init__.py
 parseforge/naming/providers/anthropic.py
+parseforge/naming/providers/azure.py
 parseforge/naming/providers/cerebras.py
 parseforge/naming/providers/cohere.py
 parseforge/naming/providers/cost.py

{parseforge-0.2.8 → parseforge-0.2.9}/parseforge.egg-info/requires.txt

@@ -7,6 +7,10 @@ textfsm-ai>=0.6.1
 anthropic>=0.40.0
 textfsm-ai[anthropic]>=0.6.1
 
+[azure]
+azure-ai-inference>=1.0.0b9
+textfsm-ai[azure]>=0.6.1
+
 [cerebras]
 openai>=1.0.0
 textfsm-ai[cerebras]>=0.6.1
@@ -29,6 +33,7 @@ anthropic>=0.40.0
 openai>=1.0.0
 mistralai==1.10.0
 cohere==5.21.1
+azure-ai-inference>=1.0.0b9
 
 [fireworks]
 openai>=1.0.0

{parseforge-0.2.8 → parseforge-0.2.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "parseforge"
-version = "0.2.8"
+version = "0.2.9"
 description = "LLM-driven pipeline that forges, validates, and promotes TextFSM templates from network CLI output"
 readme = "README.md"
 license = { text = "MIT" }
@@ -81,6 +81,7 @@ mistral = ["mistralai==1.10.0", "textfsm-ai[mistral]>=0.6.1"]
 # supporting Python 3.9 (6.x/7.x require Python >=3.10), matching
 # textfsm-ai's own pin.
 cohere = ["cohere==5.21.1", "textfsm-ai[cohere]>=0.6.1"]
+azure = ["azure-ai-inference>=1.0.0b9", "textfsm-ai[azure]>=0.6.1"]
 sampling = ["netmiko>=4.0"]
 dev = [
     "pytest>=8.0.0",
@@ -88,13 +89,15 @@ dev = [
     "ruff>=0.4.0",
     "black>=24.0.0",
     "mypy>=1.10.0",
-    # anthropic/openai/mistralai/cohere are optional at runtime (see the
-    # extras above), but every provider is exercised by the CLI/tests,
-    # which should never silently skip -- so dev always pulls all four in.
+    # anthropic/openai/mistralai/cohere/azure-ai-inference are optional at
+    # runtime (see the extras above), but every provider is exercised by
+    # the CLI/tests, which should never silently skip -- so dev always
+    # pulls all five in.
     "anthropic>=0.40.0",
     "openai>=1.0.0",
     "mistralai==1.10.0",
     "cohere==5.21.1",
+    "azure-ai-inference>=1.0.0b9",
 ]
 # Tooling for scripts/release.ps1 and CI's own package-build step.
 # bump2version isn't otherwise installed anywhere -- release.ps1 assumes

parseforge-0.2.8/parseforge/__init__.py

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