arthur-common 2.4.0__tar.gz → 2.4.1__tar.gz

{arthur_common-2.4.0 → arthur_common-2.4.1}/PKG-INFO

@@ -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 → arthur_common-2.4.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "arthur-common"
-version = "2.4.0"
+version = "2.4.1"
 description = "Utility code common to Arthur platform components."
 authors = ["Arthur <engineering@arthur.ai>"]
 license = "MIT"

{arthur_common-2.4.0 → arthur_common-2.4.1}/src/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-2.4.0 → arthur_common-2.4.1}/src/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):