arthur-common 2.4.0__py3-none-any.whl → 2.4.1__py3-none-any.whl

arthur_common/models/request_schemas.py

@@ -346,6 +346,10 @@ class PromptValidationRequest(BaseModel):
         description="The user ID this prompt belongs to",
         default=None,
     )
+    model_name: Optional[str] = Field(
+        description="The model name and version being used for this prompt (e.g., 'gpt-4', 'gpt-3.5-turbo', 'claude-3-opus', 'gemini-pro').",
+        default=None,
+    )
 
 
 class ResponseValidationRequest(BaseModel):
@@ -354,6 +358,10 @@ class ResponseValidationRequest(BaseModel):
         description="Optional data provided as context for the validation.",
         default=None,
     )
+    model_name: Optional[str] = Field(
+        description="The model name and version being used for this response (e.g., 'gpt-4', 'gpt-3.5-turbo', 'claude-3-opus', 'gemini-pro').",
+        default=None,
+    )
     # tokens: Optional[List[str]] = Field(description="optional, not used currently")
     # token_likelihoods: Optional[List[str]] = Field(
     #     description="optional, not used currently"

arthur_common/models/response_schemas.py

@@ -185,6 +185,10 @@ class ValidationResult(BaseModel):
         description="The user ID this prompt belongs to",
         default=None,
     )
+    model_name: Optional[str] = Field(
+        description="The model name and version used for this validation (e.g., 'gpt-4', 'gpt-3.5-turbo', 'claude-3-opus', 'gemini-pro').",
+        default=None,
+    )
     model_config = ConfigDict(
         json_schema_extra={
             "example": {
@@ -215,6 +219,10 @@ class ExternalInferencePrompt(BaseModel):
     message: str
     prompt_rule_results: List[ExternalRuleResult]
     tokens: int | None = None
+    model_name: Optional[str] = Field(
+        description="The model name and version used for this prompt (e.g., 'gpt-4', 'gpt-3.5-turbo', 'claude-3-opus', 'gemini-pro').",
+        default=None,
+    )
 
 
 class ExternalInferenceResponse(BaseModel):
@@ -227,6 +235,10 @@ class ExternalInferenceResponse(BaseModel):
     context: Optional[str] = None
     response_rule_results: List[ExternalRuleResult]
     tokens: int | None = None
+    model_name: Optional[str] = Field(
+        description="The model name and version used for this response (e.g., 'gpt-4', 'gpt-3.5-turbo', 'claude-3-opus', 'gemini-pro').",
+        default=None,
+    )
 
 
 class InferenceFeedbackResponse(BaseModel):
@@ -297,6 +309,10 @@ class ExternalInference(BaseModel):
     inference_response: Optional[ExternalInferenceResponse] = None
     inference_feedback: List[InferenceFeedbackResponse]
     user_id: str | None = None
+    model_name: Optional[str] = Field(
+        description="The model name and version used for this inference (e.g., 'gpt-4', 'gpt-3.5-turbo', 'claude-3-opus', 'gemini-pro').",
+        default=None,
+    )
 
 
 class QueryInferencesResponse(BaseModel):
@@ -499,6 +515,10 @@ class ChatResponse(BaseModel):
     response_results: List[ExternalRuleResult] = Field(
         description="list of rule results for the llm response",
     )
+    model_name: Optional[str] = Field(
+        description="The model name and version used for this chat response (e.g., 'gpt-4', 'gpt-3.5-turbo', 'claude-3-opus', 'gemini-pro').",
+        default=None,
+    )
 
 
 class TokenUsageCount(BaseModel):

{arthur_common-2.4.0.dist-info → arthur_common-2.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arthur-common
-Version: 2.4.0
+Version: 2.4.1
 Summary: Utility code common to Arthur platform components.
 License: MIT
 Author: Arthur

{arthur_common-2.4.0.dist-info → arthur_common-2.4.1.dist-info}/RECORD

@@ -30,8 +30,8 @@ arthur_common/models/enums.py,sha256=J2beHEMjLfOGgc-vh1aDpE7KmBGKzLoOUGYLtuciJro
 arthur_common/models/metric_schemas.py,sha256=Xf-1RTzg7iYtnBMLkUUUuMPzAujzzNvQx_pe-CksEdU,2484
 arthur_common/models/metrics.py,sha256=mCa0aN-nuNHYcqGfkyKFeriI0krz0-ScgmXWXHlKoEI,11109
 arthur_common/models/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arthur_common/models/request_schemas.py,sha256=ihrWK0SRVXsRmNaiLibbAEWi_RHl440JJvm09WRdNxQ,29329
-arthur_common/models/response_schemas.py,sha256=eZCgxnfOht8isUunAA4rosLFA-tgXRZIcj2CYa5XqOE,24362
+arthur_common/models/request_schemas.py,sha256=LEybzPoOzaaTyua48mr9sAVzrBK6dIeYhR158kMp0o8,29749
+arthur_common/models/response_schemas.py,sha256=qc6DDfY4GxtXtUiBllsQglvPnZzhe2Vw8D-S76B_r_0,25393
 arthur_common/models/schema_definitions.py,sha256=dcUSLjBmvyloStcBFmT_rHdXbKdvA8Yxi_avYUbps3E,16876
 arthur_common/models/task_job_specs.py,sha256=p7jsSb97ylHYNkwoHXNOJvx2zcnh2kxLeh3m0pddo4M,3442
 arthur_common/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -44,6 +44,6 @@ arthur_common/tools/functions.py,sha256=FWL4eWO5-vLp86WudT-MGUKvf2B8f02IdoXQFKd6
 arthur_common/tools/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arthur_common/tools/schema_inferer.py,sha256=9teI67umlGn0izp6pZ5UBuWxJthaWEmw3wRj2KPIbf4,5207
 arthur_common/tools/time_utils.py,sha256=4gfiu9NXfvPZltiVNLSIQGylX6h2W0viNi9Kv4bKyfw,1410
-arthur_common-2.4.0.dist-info/METADATA,sha256=p9N6lvjusbOqVo367KxQgD3g0mJgPMVLoxwJOAbCVRw,2146
-arthur_common-2.4.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-arthur_common-2.4.0.dist-info/RECORD,,
+arthur_common-2.4.1.dist-info/METADATA,sha256=LA7R2B8LGE78eJrwpQTHP0nFRPzLTnntZvHQUNtcVm4,2146
+arthur_common-2.4.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+arthur_common-2.4.1.dist-info/RECORD,,