unique_toolkit 0.5.55__py3-none-any.whl → 0.6.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of unique_toolkit might be problematic. Click here for more details.

Files changed (35) hide show
  1. unique_toolkit/_common/validate_required_values.py +21 -0
  2. unique_toolkit/app/__init__.py +20 -0
  3. unique_toolkit/app/schemas.py +73 -7
  4. unique_toolkit/chat/__init__.py +5 -4
  5. unique_toolkit/chat/constants.py +3 -0
  6. unique_toolkit/chat/functions.py +805 -0
  7. unique_toolkit/chat/schemas.py +11 -11
  8. unique_toolkit/chat/service.py +483 -432
  9. unique_toolkit/content/__init__.py +1 -0
  10. unique_toolkit/content/constants.py +2 -0
  11. unique_toolkit/content/functions.py +475 -0
  12. unique_toolkit/content/service.py +163 -315
  13. unique_toolkit/content/utils.py +32 -0
  14. unique_toolkit/embedding/__init__.py +3 -0
  15. unique_toolkit/embedding/constants.py +2 -0
  16. unique_toolkit/embedding/functions.py +79 -0
  17. unique_toolkit/embedding/service.py +47 -34
  18. unique_toolkit/evaluators/__init__.py +1 -0
  19. unique_toolkit/evaluators/constants.py +1 -0
  20. unique_toolkit/evaluators/context_relevancy/constants.py +3 -3
  21. unique_toolkit/evaluators/context_relevancy/utils.py +5 -2
  22. unique_toolkit/evaluators/hallucination/utils.py +2 -1
  23. unique_toolkit/language_model/__init__.py +1 -0
  24. unique_toolkit/language_model/constants.py +4 -0
  25. unique_toolkit/language_model/functions.py +229 -0
  26. unique_toolkit/language_model/service.py +76 -343
  27. unique_toolkit/short_term_memory/__init__.py +5 -0
  28. unique_toolkit/short_term_memory/constants.py +1 -0
  29. unique_toolkit/short_term_memory/functions.py +175 -0
  30. unique_toolkit/short_term_memory/service.py +153 -27
  31. {unique_toolkit-0.5.55.dist-info → unique_toolkit-0.6.1.dist-info}/METADATA +38 -8
  32. unique_toolkit-0.6.1.dist-info/RECORD +64 -0
  33. unique_toolkit-0.5.55.dist-info/RECORD +0 -50
  34. {unique_toolkit-0.5.55.dist-info → unique_toolkit-0.6.1.dist-info}/LICENSE +0 -0
  35. {unique_toolkit-0.5.55.dist-info → unique_toolkit-0.6.1.dist-info}/WHEEL +0 -0
unique_toolkit/chat/schemas.py CHANGED
@@ -69,34 +69,34 @@ class ChatMessage(BaseModel):
69
69
  return self
70
70
 
71
71
 
72
- class MessageAssessmentStatus(StrEnum):
72
+ class ChatMessageAssessmentStatus(StrEnum):
73
73
  PENDING = "PENDING"
74
74
  DONE = "DONE"
75
75
  ERROR = "ERROR"
76
76
 
77
77
 
78
- class MessageAssessmentLabel(StrEnum):
79
- POSITIVE = "POSITIVE"
80
- NEGATIVE = "NEGATIVE"
81
- VERIFIED = "VERIFIED"
82
- UNVERIFIED = "UNVERIFIED"
78
+ class ChatMessageAssessmentLabel(StrEnum):
79
+ RED = "RED"
80
+ YELLOW = "YELLOW"
81
+ GREEN = "GREEN"
83
82
 
84
83
 
85
- class MessageAssessmentType(StrEnum):
84
+ class ChatMessageAssessmentType(StrEnum):
86
85
  HALLUCINATION = "HALLUCINATION"
87
86
  COMPLIANCE = "COMPLIANCE"
88
87
 
89
88
 
90
- class MessageAssessment(BaseModel):
89
+ class ChatMessageAssessment(BaseModel):
91
90
  model_config = model_config
92
91
 
93
92
  id: str
94
93
  object: str
95
94
  message_id: str
96
- status: MessageAssessmentStatus
97
- type: MessageAssessmentType
95
+ status: ChatMessageAssessmentStatus
96
+ type: ChatMessageAssessmentType
97
+ title: str | None = None
98
98
  explanation: str | None = None
99
- label: MessageAssessmentLabel | None = None
99
+ label: ChatMessageAssessmentLabel | None = None
100
100
  is_visible: bool
101
101
  created_at: datetime | None = None
102
102
  updated_at: datetime | None = None