dtx-models 0.18.2__py3-none-any.whl → 0.19.0__py3-none-any.whl

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/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/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

{dtx_models-0.18.2.dist-info → dtx_models-0.19.0.dist-info}/METADATA

@@ -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.dist-info → dtx_models-0.19.0.dist-info}/RECORD

@@ -1,9 +1,9 @@
 dtx_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dtx_models/analysis.py,sha256=hrCy24jwS9uCXz7h6fTcAxgw5u_Stz9zsSZywCdQYCs,11539
+dtx_models/analysis.py,sha256=2cOFSVFOJ2JMNaAxK5aaMMvbg6CLe7jhlnJCSMx0b7E,11547
 dtx_models/base.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dtx_models/evaluator.py,sha256=h1XDCrvbVoqkexNBfc4fiUkXNAlq8DhpMoybVmAgjYU,10026
 dtx_models/exceptions.py,sha256=gxzdkoiZwsjfWKIT2m3hHM1FNO1l3gDj979rDUhKt_g,41
-dtx_models/prompts.py,sha256=tXCvdkuukjs_3j4lS1R_MSq_WOconmiWtFzLYdULocY,15164
+dtx_models/prompts.py,sha256=l0JR1CI03wcVp-Ty6lvujS3eIIJVUJnF-Tj0bOt0yT8,15172
 dtx_models/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dtx_models/providers/base.py,sha256=KYEhVF61INjsGxO5tKOENv1c1xGGb1kX6Jevpewf5UI,418
 dtx_models/providers/gradio.py,sha256=2edCLtTAILvJkiMQkYl4FYKixMPjHrRUB0OFK9SeuIU,5261
@@ -15,7 +15,7 @@ dtx_models/providers/models_spec.py,sha256=pc8yCITiaErZSp8H9f49fSBZEeEp2zPYfFN4Q
 dtx_models/providers/ollama.py,sha256=VfCAOO9YTPMLUbcsiKXDQgG_SYSGVUsCfS7iwfhCogk,3207
 dtx_models/providers/openai.py,sha256=psxmTeOdK5g2sOyLTBRTsxtgtBudXAWLCvzH8CeTNnw,3671
 dtx_models/results.py,sha256=sah7v1jOXAd8Vz_ghlamgTSaYxjvI8GdBzpLq0O2OJU,4323
-dtx_models/scope.py,sha256=PwmoOS6i4FLTKxklUH8Ueb633RH_l3_6x3ppA4hUmfk,7071
+dtx_models/scope.py,sha256=rzB6cQhH5zjmL5F1m--_Jc7bxAYkY5AHB7x3tXcEcWw,7079
 dtx_models/tactic.py,sha256=GFSqgRmf00P6qJKKuLo14h3Tg-TRkl9nKk7oLvzDMdI,1395
 dtx_models/target.py,sha256=EETY2MzhWTuS4Xc8DmAQqrnesSuCttPaYTkzzt2Wn8M,5803
 dtx_models/template/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -24,6 +24,6 @@ dtx_models/template/prompts/base.py,sha256=E1ygnImn7RBY1puRhnyz3Ftz77PgahipqG2Yt
 dtx_models/template/prompts/langhub.py,sha256=90qvLl94hxg4mRRfhanoDICfS1YQyy3ply1ZYre7l6I,2414
 dtx_models/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dtx_models/utils/urls.py,sha256=1uuarlhCBi-BVTPvtvCz7l7kM-b7rVCmPI_9ktHrKYw,816
-dtx_models-0.18.2.dist-info/METADATA,sha256=qMlSzJrB_Z_nfMPDWgo2y3uoajrCvGFJAhjwBOdU6kw,1484
-dtx_models-0.18.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dtx_models-0.18.2.dist-info/RECORD,,
+dtx_models-0.19.0.dist-info/METADATA,sha256=Uo5BFyRhrHhlVj_aXIDv94j5a2_3hEy1qtT3LVuu1DI,1484
+dtx_models-0.19.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dtx_models-0.19.0.dist-info/RECORD,,