dtx-models 0.18.2__tar.gz → 0.19.0__tar.gz

{dtx_models-0.18.2 → dtx_models-0.19.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dtx-models
-Version: 0.18.2
+Version: 0.19.0
 Summary: Shared model schemas and YAML-based configurations for the DTX framework.
 Author: JC
 Author-email: jitendra@detoxio.ai

{dtx_models-0.18.2 → dtx_models-0.19.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "dtx-models"
-version = "0.18.2"
+version = "0.19.0"
 description = "Shared model schemas and YAML-based configurations for the DTX framework."
 authors = ["JC <jitendra@detoxio.ai>"]
 readme = "README.md"

{dtx_models-0.18.2 → dtx_models-0.19.0}/src/dtx_models/analysis.py

@@ -6,7 +6,7 @@ from pydantic import BaseModel, ConfigDict, Field, field_serializer, field_valid
 
 from .evaluator import BasePromptEvaluation
 from .prompts import BaseTestStrPrompt, MultiturnTestPrompts
-from .scope import Agent, RedTeamScope
+from .scope import AgentInfo, RedTeamScope
 
 # ----------------------
 # Define Models
@@ -92,7 +92,7 @@ class ThreatModel(BaseModel):
     analysis: str = Field(
         description="Thinking and analysis performed solve the problem as approach "
     )
-    target: Agent = Field(
+    target: AgentInfo = Field(
         description="Target agent with necessary architectural details"
     )
     threat_actors: List[str] = Field(description="Potential Threat Actors")

{dtx_models-0.18.2 → dtx_models-0.19.0}/src/dtx_models/prompts.py

@@ -286,7 +286,7 @@ class MultiTurnTestPrompt(BaseMultiTurnConversation):
     )
     base_prompt: str = Field(
         default="",
-        description="Base prompt in its most simplistic form that need to be answered by AI Agent. Generally it is the harmful.",
+        description="Base prompt in its most simplistic form that need to be answered by AI AgentInfo. Generally it is the harmful.",
     )
 
     model_config = ConfigDict(frozen=True)  # Make fields immutable
@@ -328,7 +328,7 @@ class BaseMultiTurnAgentResponse(BaseMultiTurnResponse):
 
     response: Optional[Union[str, List[Any], Dict[str, Any]]] = Field(
         default_factory=dict,
-        description="Final Agent Response",
+        description="Final AgentInfo Response",
     )
 
     scores: Optional[Dict[str, float]] = Field(

{dtx_models-0.18.2 → dtx_models-0.19.0}/src/dtx_models/scope.py

@@ -16,7 +16,7 @@ from .template.prompts.base import PromptsRepoType
 from .template.prompts.langhub import LangHubPromptTemplate
 
 
-class Agent(BaseModel):
+class AgentInfo(BaseModel):
     """Represents an AI agent with various integrations, capabilities, and restrictions."""
 
     name: Optional[str] = Field(default_factory=lambda: "")  # Default value for name
@@ -204,5 +204,5 @@ class ProvidersWithEnvironments(BaseModel):
 class RedTeamScope(ProvidersWithEnvironments):
     """Represents a red teaming scope, which includes an agent and its related details."""
 
-    agent: Agent
+    agent: AgentInfo
     redteam: RedTeamSettings