parseforge 0.2.4__tar.gz → 0.2.5__tar.gz

{parseforge-0.2.4/parseforge.egg-info → parseforge-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: parseforge
-Version: 0.2.4
+Version: 0.2.5
 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>
@@ -33,6 +33,9 @@ Requires-Dist: textfsm-ai[anthropic]>=0.6.1; extra == "anthropic"
 Provides-Extra: deepseek
 Requires-Dist: openai>=1.0.0; extra == "deepseek"
 Requires-Dist: textfsm-ai[deepseek]>=0.6.1; extra == "deepseek"
+Provides-Extra: openai
+Requires-Dist: openai>=1.0.0; extra == "openai"
+Requires-Dist: textfsm-ai[openai]>=0.6.1; extra == "openai"
 Provides-Extra: sampling
 Requires-Dist: netmiko>=4.0; extra == "sampling"
 Provides-Extra: dev
@@ -83,11 +86,11 @@ pip install parseforge[anthropic]
 command that's pure local processing (`canonical`/`readable`/`recognizers`,
 `integration`, `promotion`) works with nothing further. Anything that calls an
 LLM (`name`, `check --provider`, `run`, `generate-template`, `trial`) needs the
-extra for whichever provider it uses: `anthropic` or `deepseek`. `--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,deepseek,sampling]`.
+extra for whichever provider it uses: `anthropic`, `openai`, or `deepseek`.
+`--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,sampling]`.
 
 ## Development
 
@@ -95,9 +98,10 @@ sampling; combine extras as needed, e.g.
 pip install -e ".[dev,sampling]"
 pytest
 ```
-`dev` already includes both `anthropic` and `openai` (tests exercise both
-providers, never silently skip) — add `,anthropic`/`,deepseek` explicitly only
-if installing outside of `dev`.
+`dev` already includes both the `anthropic` and `openai` SDKs (tests exercise
+all three providers — `anthropic`, `openai`, `deepseek` share just those two
+packages — and never silently skip) — add `,anthropic`/`,openai`/`,deepseek`
+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
@@ -119,8 +123,8 @@ 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`/
-`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`). A cache hit (a command already seen before)
+never touches the LLM, so no key is needed at all in that case.
 
 **`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

{parseforge-0.2.4 → parseforge-0.2.5}/README.md

@@ -33,11 +33,11 @@ pip install parseforge[anthropic]
 command that's pure local processing (`canonical`/`readable`/`recognizers`,
 `integration`, `promotion`) works with nothing further. Anything that calls an
 LLM (`name`, `check --provider`, `run`, `generate-template`, `trial`) needs the
-extra for whichever provider it uses: `anthropic` or `deepseek`. `--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,deepseek,sampling]`.
+extra for whichever provider it uses: `anthropic`, `openai`, or `deepseek`.
+`--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,sampling]`.
 
 ## Development
 
@@ -45,9 +45,10 @@ sampling; combine extras as needed, e.g.
 pip install -e ".[dev,sampling]"
 pytest
 ```
-`dev` already includes both `anthropic` and `openai` (tests exercise both
-providers, never silently skip) — add `,anthropic`/`,deepseek` explicitly only
-if installing outside of `dev`.
+`dev` already includes both the `anthropic` and `openai` SDKs (tests exercise
+all three providers — `anthropic`, `openai`, `deepseek` share just those two
+packages — and never silently skip) — add `,anthropic`/`,openai`/`,deepseek`
+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
@@ -69,8 +70,8 @@ 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`/
-`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`). A cache hit (a command already seen before)
+never touches the LLM, so no key is needed at all in that case.
 
 **`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

parseforge-0.2.5/parseforge/__init__.py

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

{parseforge-0.2.4 → parseforge-0.2.5}/parseforge/cli/main.py

@@ -24,6 +24,7 @@ from parseforge.cli import config as cli_config
 _BUILDERS: dict[str, type[naming.RegexBuilder]] = {
     "anthropic": naming.AnthropicRegexBuilder,
     "deepseek": naming.DeepSeekRegexBuilder,
+    "openai": naming.OpenAIRegexBuilder,
 }
 
 _CONNECTORS = ("netmiko",)
@@ -198,7 +199,8 @@ def main() -> None:
     "--api-key",
     default=None,
     help="Provider API key. Defaults to that provider's own API key environment "
-    "variable (ANTHROPIC_API_KEY, DEEPSEEK_API_KEY); only needed on a cache miss.",
+    "variable (ANTHROPIC_API_KEY, OPENAI_API_KEY, DEEPSEEK_API_KEY); only needed "
+    "on a cache miss.",
 )
 @click.option(
     "--model",
@@ -266,7 +268,7 @@ def name_cmd(
     "--provider",
     required=True,
     help="LLM provider for template generation (textfsm-ai's own registry, "
-    'e.g. "anthropic", "deepseek"). Naming uses its own separate '
+    'e.g. "anthropic", "openai", "deepseek"). Naming uses its own separate '
     "--naming-provider, not this one.",
 )
 @click.option("--api-key", required=True, help="API key for the generation LLM call.")

{parseforge-0.2.4 → parseforge-0.2.5}/parseforge/naming/__init__.py

@@ -8,7 +8,7 @@ from .llm import (
     UnimplementedRegexBuilder,
     build_prompt,
 )
-from .providers import AnthropicRegexBuilder, DeepSeekRegexBuilder
+from .providers import AnthropicRegexBuilder, DeepSeekRegexBuilder, OpenAIRegexBuilder
 from .resolver import NamingResolution, cli_name, resolve_cli_name
 
 __all__ = [
@@ -20,6 +20,7 @@ __all__ = [
     "UnimplementedRegexBuilder",
     "AnthropicRegexBuilder",
     "DeepSeekRegexBuilder",
+    "OpenAIRegexBuilder",
     "LLMCLIResponse",
     "TokenUsage",
     "build_prompt",

parseforge-0.2.5/parseforge/naming/providers/__init__.py

@@ -0,0 +1,5 @@
+from .anthropic import AnthropicRegexBuilder
+from .deepseek import DeepSeekRegexBuilder
+from .openai import OpenAIRegexBuilder
+
+__all__ = ["AnthropicRegexBuilder", "DeepSeekRegexBuilder", "OpenAIRegexBuilder"]

{parseforge-0.2.4 → parseforge-0.2.5}/parseforge/naming/providers/models.yaml

@@ -22,3 +22,12 @@ deepseek:
     - deepseek-v4-flash
     - deepseek-v4-pro
   deprecated: []
+
+openai:
+  default: gpt-5.4-mini
+  supported:
+    - gpt-5.4-mini
+    - gpt-5.4
+    - gpt-5.4-nano
+    - gpt-5.5
+  deprecated: []

parseforge-0.2.5/parseforge/naming/providers/openai.py

@@ -0,0 +1,117 @@
+"""OpenAI-backed RegexBuilder implementation."""
+
+from __future__ import annotations
+
+import time
+from typing import TYPE_CHECKING, Any
+
+from ..llm import CliContext, LLMCLIResponse, TokenUsage, build_prompt
+from .cost import estimate_cost
+from .errors import format_llm_error_reason, is_retryable
+from .models import default_model
+from .text import extract_pattern
+
+if TYPE_CHECKING:
+    import openai as openai_sdk
+
+DEFAULT_MODEL = default_model("openai")
+
+_DEFAULT_MAX_TOKENS = 1024
+
+
+def _import_openai() -> Any:
+    """Deferred import — openai is an optional extra (parseforge[openai]);
+    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:
+        import openai
+    except ImportError as exc:
+        raise ImportError(
+            "the openai package is required to use OpenAIRegexBuilder — "
+            "install it via `pip install parseforge[openai]`"
+        ) from exc
+    return openai
+
+
+class OpenAIRegexBuilder:
+    """Builds a cli-name regex pattern by prompting an OpenAI model.
+
+    The client is constructed lazily, on the first actual call — not in
+    ``__init__`` — so this can be used as a default RegexBuilder without
+    requiring ``OPENAI_API_KEY`` to be set for cache-hit lookups, which
+    never reach the LLM at all (see resolver.cli_name).
+
+    API key resolves from the ``api_key`` argument if given, otherwise the
+    ``OPENAI_API_KEY`` environment variable (the SDK's own default).
+    """
+
+    provider = "openai"
+
+    def __init__(self, model: str = DEFAULT_MODEL, api_key: str | None = None) -> None:
+        self.model = model
+        self._api_key = api_key
+        self._client: openai_sdk.OpenAI | None = None
+
+    def _get_client(self) -> openai_sdk.OpenAI:
+        if self._client is None:
+            openai = _import_openai()
+            self._client = (
+                openai.OpenAI(api_key=self._api_key)
+                if self._api_key
+                else openai.OpenAI()
+            )
+        return self._client
+
+    def build_pattern(
+        self, command: str, context: CliContext, **kwargs: Any
+    ) -> LLMCLIResponse:
+        openai = _import_openai()
+        prompt = build_prompt(command, context)
+        max_tokens = kwargs.pop("max_tokens", None) or _DEFAULT_MAX_TOKENS
+
+        start = time.monotonic()
+        try:
+            response = self._get_client().chat.completions.create(
+                model=self.model,
+                max_tokens=max_tokens,
+                messages=[{"role": "user", "content": prompt}],
+                **kwargs,
+            )
+        except openai.OpenAIError 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_llm_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,
+                    model=self.model,
+                ),
+            ),
+            duration_ms=duration_ms,
+            reason=choice.finish_reason or "",
+            ready=choice.finish_reason == "stop",
+        )

{parseforge-0.2.4 → parseforge-0.2.5/parseforge.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: parseforge
-Version: 0.2.4
+Version: 0.2.5
 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>
@@ -33,6 +33,9 @@ Requires-Dist: textfsm-ai[anthropic]>=0.6.1; extra == "anthropic"
 Provides-Extra: deepseek
 Requires-Dist: openai>=1.0.0; extra == "deepseek"
 Requires-Dist: textfsm-ai[deepseek]>=0.6.1; extra == "deepseek"
+Provides-Extra: openai
+Requires-Dist: openai>=1.0.0; extra == "openai"
+Requires-Dist: textfsm-ai[openai]>=0.6.1; extra == "openai"
 Provides-Extra: sampling
 Requires-Dist: netmiko>=4.0; extra == "sampling"
 Provides-Extra: dev
@@ -83,11 +86,11 @@ pip install parseforge[anthropic]
 command that's pure local processing (`canonical`/`readable`/`recognizers`,
 `integration`, `promotion`) works with nothing further. Anything that calls an
 LLM (`name`, `check --provider`, `run`, `generate-template`, `trial`) needs the
-extra for whichever provider it uses: `anthropic` or `deepseek`. `--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,deepseek,sampling]`.
+extra for whichever provider it uses: `anthropic`, `openai`, or `deepseek`.
+`--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,sampling]`.
 
 ## Development
 
@@ -95,9 +98,10 @@ sampling; combine extras as needed, e.g.
 pip install -e ".[dev,sampling]"
 pytest
 ```
-`dev` already includes both `anthropic` and `openai` (tests exercise both
-providers, never silently skip) — add `,anthropic`/`,deepseek` explicitly only
-if installing outside of `dev`.
+`dev` already includes both the `anthropic` and `openai` SDKs (tests exercise
+all three providers — `anthropic`, `openai`, `deepseek` share just those two
+packages — and never silently skip) — add `,anthropic`/`,openai`/`,deepseek`
+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
@@ -119,8 +123,8 @@ 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`/
-`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`). A cache hit (a command already seen before)
+never touches the LLM, so no key is needed at all in that case.
 
 **`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

{parseforge-0.2.4 → parseforge-0.2.5}/parseforge.egg-info/SOURCES.txt

@@ -34,6 +34,7 @@ parseforge/naming/providers/deepseek.py
 parseforge/naming/providers/errors.py
 parseforge/naming/providers/models.py
 parseforge/naming/providers/models.yaml
+parseforge/naming/providers/openai.py
 parseforge/naming/providers/text.py
 parseforge/sampling/__init__.py
 parseforge/sampling/core.py

{parseforge-0.2.4 → parseforge-0.2.5}/parseforge.egg-info/requires.txt

@@ -20,6 +20,10 @@ mypy>=1.10.0
 anthropic>=0.40.0
 openai>=1.0.0
 
+[openai]
+openai>=1.0.0
+textfsm-ai[openai]>=0.6.1
+
 [release]
 bump2version
 build

{parseforge-0.2.4 → parseforge-0.2.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "parseforge"
-version = "0.2.4"
+version = "0.2.5"
 description = "LLM-driven pipeline that forges, validates, and promotes TextFSM templates from network CLI output"
 readme = "README.md"
 license = { text = "MIT" }
@@ -64,6 +64,7 @@ keywords = [
 # provider(s) you actually use.
 anthropic = ["anthropic>=0.40.0", "textfsm-ai[anthropic]>=0.6.1"]
 deepseek = ["openai>=1.0.0", "textfsm-ai[deepseek]>=0.6.1"]
+openai = ["openai>=1.0.0", "textfsm-ai[openai]>=0.6.1"]
 sampling = ["netmiko>=4.0"]
 dev = [
     "pytest>=8.0.0",

parseforge-0.2.4/parseforge/__init__.py

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

parseforge-0.2.4/parseforge/naming/providers/__init__.py

@@ -1,4 +0,0 @@
-from .anthropic import AnthropicRegexBuilder
-from .deepseek import DeepSeekRegexBuilder
-
-__all__ = ["AnthropicRegexBuilder", "DeepSeekRegexBuilder"]