prompt-firewall 0.1.4__tar.gz → 0.1.5__tar.gz

{prompt_firewall-0.1.4 → prompt_firewall-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompt-firewall
-Version: 0.1.4
+Version: 0.1.5
 Summary: Prompt guard library for composing multiple safety checks.
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
@@ -22,8 +22,6 @@ Composable prompt-safety guards as a lightweight Python library. It provides:
 from prompt_guard import GuardPipeline, PromptGuardConfig
 
 cfg = PromptGuardConfig(
-    phrases_path="phrases.txt",
-    sentences_path="sentences.txt",
     embed_model_name="mxbai-embed-large",
     judge_model_name="qwen2.5:3b-instruct",
 )

{prompt_firewall-0.1.4 → prompt_firewall-0.1.5}/README.md

@@ -11,8 +11,6 @@ Composable prompt-safety guards as a lightweight Python library. It provides:
 from prompt_guard import GuardPipeline, PromptGuardConfig
 
 cfg = PromptGuardConfig(
-    phrases_path="phrases.txt",
-    sentences_path="sentences.txt",
     embed_model_name="mxbai-embed-large",
     judge_model_name="qwen2.5:3b-instruct",
 )

{prompt_firewall-0.1.4 → prompt_firewall-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "prompt-firewall"
-version = "0.1.4"
+version = "0.1.5"
 description = "Prompt guard library for composing multiple safety checks."
 readme = "README.md"
 requires-python = ">=3.13"

{prompt_firewall-0.1.4 → prompt_firewall-0.1.5}/src/prompt_firewall.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompt-firewall
-Version: 0.1.4
+Version: 0.1.5
 Summary: Prompt guard library for composing multiple safety checks.
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
@@ -22,8 +22,6 @@ Composable prompt-safety guards as a lightweight Python library. It provides:
 from prompt_guard import GuardPipeline, PromptGuardConfig
 
 cfg = PromptGuardConfig(
-    phrases_path="phrases.txt",
-    sentences_path="sentences.txt",
     embed_model_name="mxbai-embed-large",
     judge_model_name="qwen2.5:3b-instruct",
 )

prompt_firewall-0.1.5/src/prompt_guard/config.py

@@ -0,0 +1,77 @@
+from pydantic import BaseModel, Field
+
+
+BUILTIN_DEFAULT_PHRASES_RESOURCE = "data/phrases.txt"
+BUILTIN_DEFAULT_SENTENCES_RESOURCE = "data/sentences.txt"
+
+
+class PromptGuardConfig(BaseModel):
+    phrases_path: str | None = Field(
+        default=None,
+        examples=["/path/to/phrases.txt"],
+        description="Path to phrases file (one phrase per line).",
+    )
+    sentences_path: str | None = Field(
+        default=None,
+        examples=["/path/to/sentences.txt"],
+        description="Path to sentences file (one sentence per line).",
+    )
+
+    tfidf_top_k: int = Field(
+        default=5,
+        examples=[5],
+        description="Number of top TF-IDF matches to keep.",
+    )
+    tfidf_ngram_range: tuple[int, int] = Field(
+        default=(1, 3),
+        examples=[(1, 3)],
+        description="N-gram range for TF-IDF vectorizer.",
+    )
+    rag_top_k: int = Field(
+        default=5,
+        examples=[5],
+        description="Number of top RAG matches to retrieve.",
+    )
+
+    embed_model_name: str = Field(
+        default="mxbai-embed-large",
+        examples=["mxbai-embed-large"],
+        description="Ollama embedding model name.",
+    )
+    judge_model_name: str = Field(
+        default="qwen2.5:3b-instruct",
+        examples=["qwen2.5:3b-instruct"],
+        description="Ollama judge model name.",
+    )
+    judge_temperature: float = Field(
+        default=0.0,
+        examples=[0.0],
+        description="LLM judge temperature.",
+    )
+    judge_max_tokens: int = Field(
+        default=256,
+        examples=[256],
+        description="LLM judge max tokens.",
+    )
+
+    base_url: str | None = Field(
+        default=None,
+        examples=["http://localhost:11434"],
+        description="Optional base URL for Ollama.",
+    )
+
+    enable_tfidf: bool = Field(
+        default=True,
+        examples=[True],
+        description="Enable TF-IDF guard.",
+    )
+    enable_rag: bool = Field(
+        default=True,
+        examples=[True],
+        description="Enable RAG guard.",
+    )
+    enable_judge: bool = Field(
+        default=True,
+        examples=[True],
+        description="Enable LLM judge guard.",
+    )

{prompt_firewall-0.1.4 → prompt_firewall-0.1.5}/src/prompt_guard/pipeline.py

@@ -6,8 +6,8 @@ from llama_index.embeddings.ollama import OllamaEmbedding
 from sklearn.feature_extraction.text import TfidfVectorizer
 
 from .config import (
-    DEFAULT_PHRASES_RESOURCE,
-    DEFAULT_SENTENCES_RESOURCE,
+    BUILTIN_DEFAULT_PHRASES_RESOURCE,
+    BUILTIN_DEFAULT_SENTENCES_RESOURCE,
     PromptGuardConfig,
 )
 from .guards.judge_guard import LlmJudgeGuard
@@ -43,7 +43,7 @@ class GuardPipeline:
     def _build_tfidf_guard(self) -> TfIdfGuard:
         phrases = self._load_lines(
             self._config.phrases_path,
-            default_resource=DEFAULT_PHRASES_RESOURCE,
+            default_resource=BUILTIN_DEFAULT_PHRASES_RESOURCE,
         )
         vectorizer = TfidfVectorizer(ngram_range=self._config.tfidf_ngram_range)
         phrase_matrix = vectorizer.fit_transform(phrases)
@@ -57,7 +57,7 @@ class GuardPipeline:
     def _build_rag_guard(self) -> RagGuard:
         lines = self._load_lines(
             self._config.sentences_path,
-            default_resource=DEFAULT_SENTENCES_RESOURCE,
+            default_resource=BUILTIN_DEFAULT_SENTENCES_RESOURCE,
         )
         docs = [Document(text=line) for line in lines]
         embed_kwargs = {"model_name": self._config.embed_model_name}

prompt_firewall-0.1.4/src/prompt_guard/config.py

@@ -1,30 +0,0 @@
-from dataclasses import dataclass
-
-
-DEFAULT_PHRASES_RESOURCE = "data/phrases.txt"
-DEFAULT_SENTENCES_RESOURCE = "data/sentences.txt"
-
-
-@dataclass(slots=True)
-class PromptGuardConfig:
-    phrases_path: str | None = None
-    sentences_path: str | None = None
-
-    tfidf_top_k: int = 5
-    tfidf_ngram_range: tuple[int, int] = (1, 3)
-    rag_top_k: int = 5
-
-    embed_model_name: str = "mxbai-embed-large"
-    judge_model_name: str = "qwen2.5:3b-instruct"
-    judge_temperature: float = 0.0
-    judge_max_tokens: int = 256
-
-    base_url: str | None = None
-
-    tfidf_limit: float = 0.0
-    rag_limit: float = 0.0
-    judge_limit: float = 0.0
-
-    enable_tfidf: bool = True
-    enable_rag: bool = True
-    enable_judge: bool = True