judgeval 0.16.0__py3-none-any.whl → 0.16.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.
Files changed (40) hide show
  1. judgeval/api/api_types.py +2 -1
  2. judgeval/data/judgment_types.py +2 -1
  3. judgeval/logger.py +1 -1
  4. judgeval/tracer/__init__.py +10 -7
  5. judgeval/tracer/keys.py +7 -3
  6. judgeval/tracer/llm/__init__.py +2 -1259
  7. judgeval/tracer/llm/config.py +110 -0
  8. judgeval/tracer/llm/constants.py +10 -0
  9. judgeval/tracer/llm/llm_anthropic/__init__.py +3 -0
  10. judgeval/tracer/llm/llm_anthropic/wrapper.py +611 -0
  11. judgeval/tracer/llm/llm_google/__init__.py +0 -0
  12. judgeval/tracer/llm/llm_google/config.py +24 -0
  13. judgeval/tracer/llm/llm_google/wrapper.py +426 -0
  14. judgeval/tracer/llm/llm_groq/__init__.py +0 -0
  15. judgeval/tracer/llm/llm_groq/config.py +23 -0
  16. judgeval/tracer/llm/llm_groq/wrapper.py +477 -0
  17. judgeval/tracer/llm/llm_openai/__init__.py +3 -0
  18. judgeval/tracer/llm/llm_openai/wrapper.py +637 -0
  19. judgeval/tracer/llm/llm_together/__init__.py +0 -0
  20. judgeval/tracer/llm/llm_together/config.py +23 -0
  21. judgeval/tracer/llm/llm_together/wrapper.py +478 -0
  22. judgeval/tracer/llm/providers.py +5 -5
  23. judgeval/tracer/processors/__init__.py +1 -1
  24. judgeval/trainer/console.py +1 -1
  25. judgeval/utils/decorators/__init__.py +0 -0
  26. judgeval/utils/decorators/dont_throw.py +21 -0
  27. judgeval/utils/{decorators.py → decorators/use_once.py} +0 -11
  28. judgeval/utils/meta.py +1 -1
  29. judgeval/utils/version_check.py +1 -1
  30. judgeval/version.py +1 -1
  31. {judgeval-0.16.0.dist-info → judgeval-0.16.1.dist-info}/METADATA +1 -1
  32. {judgeval-0.16.0.dist-info → judgeval-0.16.1.dist-info}/RECORD +37 -23
  33. judgeval/tracer/llm/google/__init__.py +0 -21
  34. judgeval/tracer/llm/groq/__init__.py +0 -20
  35. judgeval/tracer/llm/together/__init__.py +0 -20
  36. /judgeval/tracer/llm/{anthropic/__init__.py → llm_anthropic/config.py} +0 -0
  37. /judgeval/tracer/llm/{openai/__init__.py → llm_openai/config.py} +0 -0
  38. {judgeval-0.16.0.dist-info → judgeval-0.16.1.dist-info}/WHEEL +0 -0
  39. {judgeval-0.16.0.dist-info → judgeval-0.16.1.dist-info}/entry_points.txt +0 -0
  40. {judgeval-0.16.0.dist-info → judgeval-0.16.1.dist-info}/licenses/LICENSE.md +0 -0
judgeval/api/api_types.py CHANGED
@@ -1,6 +1,6 @@
1
1
  # generated by datamodel-codegen:
2
2
  # filename: .openapi.json
3
- # timestamp: 2025-10-07T20:43:52+00:00
3
+ # timestamp: 2025-10-09T00:16:42+00:00
4
4
 
5
5
  from __future__ import annotations
6
6
  from typing import Any, Dict, List, Literal, Optional, TypedDict, Union
@@ -94,6 +94,7 @@ class ResolveProjectNameRequest(TypedDict):
94
94
 
95
95
  class ResolveProjectNameResponse(TypedDict):
96
96
  project_id: str
97
+ project_created: bool
97
98
 
98
99
 
99
100
  class TraceIdRequest(TypedDict):
@@ -1,6 +1,6 @@
1
1
  # generated by datamodel-codegen:
2
2
  # filename: .openapi.json
3
- # timestamp: 2025-10-07T20:43:51+00:00
3
+ # timestamp: 2025-10-09T00:16:41+00:00
4
4
 
5
5
  from __future__ import annotations
6
6
  from typing import Annotated, Any, Dict, List, Optional, Union
@@ -101,6 +101,7 @@ class ResolveProjectNameRequest(BaseModel):
101
101
 
102
102
  class ResolveProjectNameResponse(BaseModel):
103
103
  project_id: Annotated[str, Field(title="Project Id")]
104
+ project_created: Annotated[bool, Field(title="Project Created")]
104
105
 
105
106
 
106
107
  class TraceIdRequest(BaseModel):
judgeval/logger.py CHANGED
@@ -2,7 +2,7 @@ import logging
2
2
  import sys
3
3
 
4
4
  from judgeval.env import JUDGMENT_NO_COLOR
5
- from judgeval.utils.decorators import use_once
5
+ from judgeval.utils.decorators.use_once import use_once
6
6
 
7
7
  RESET = "\033[0m"
8
8
  RED = "\033[31m"
@@ -55,7 +55,7 @@ from judgeval.tracer.managers import (
55
55
  sync_agent_context,
56
56
  async_agent_context,
57
57
  )
58
- from judgeval.utils.decorators import dont_throw
58
+ from judgeval.utils.decorators.dont_throw import dont_throw
59
59
  from judgeval.utils.guards import expect_api_key, expect_organization_id
60
60
  from judgeval.utils.serialize import safe_serialize
61
61
  from judgeval.utils.meta import SingletonMeta
@@ -159,11 +159,14 @@ class Tracer(metaclass=SingletonMeta):
159
159
 
160
160
  self.judgment_processor = NoOpJudgmentSpanProcessor()
161
161
  if self.enable_monitoring:
162
- project_id = Tracer._resolve_project_id(
162
+ project_id, project_created = Tracer._resolve_project_id(
163
163
  self.project_name, self.api_key, self.organization_id
164
- )
165
-
164
+ ) or (None, False)
166
165
  if project_id:
166
+ if project_created:
167
+ judgeval_logger.info(
168
+ f"Project {self.project_name} was autocreated successfully."
169
+ )
167
170
  self.judgment_processor = self.get_processor(
168
171
  tracer=self,
169
172
  project_name=self.project_name,
@@ -179,7 +182,7 @@ class Tracer(metaclass=SingletonMeta):
179
182
  set_tracer_provider(provider)
180
183
  else:
181
184
  judgeval_logger.error(
182
- f"Failed to resolve project {self.project_name}, please create it first at https://app.judgmentlabs.ai/org/{self.organization_id}/projects. Skipping Judgment export."
185
+ f"Failed to resolve or autocreate project {self.project_name}, please create it first at https://app.judgmentlabs.ai/org/{self.organization_id}/projects. Skipping Judgment export."
183
186
  )
184
187
 
185
188
  self.tracer = get_tracer_provider().get_tracer(
@@ -237,14 +240,14 @@ class Tracer(metaclass=SingletonMeta):
237
240
  @staticmethod
238
241
  def _resolve_project_id(
239
242
  project_name: str, api_key: str, organization_id: str
240
- ) -> str | None:
243
+ ) -> Tuple[str, bool]:
241
244
  """Resolve project_id from project_name using the API."""
242
245
  client = JudgmentSyncClient(
243
246
  api_key=api_key,
244
247
  organization_id=organization_id,
245
248
  )
246
249
  response = client.projects_resolve({"project_name": project_name})
247
- return response["project_id"]
250
+ return response["project_id"], response["project_created"]
248
251
 
249
252
  def get_current_span(self):
250
253
  return get_current_span()
judgeval/tracer/keys.py CHANGED
@@ -12,6 +12,8 @@ class AttributeKeys(str, Enum):
12
12
  JUDGMENT_OFFLINE_MODE = "judgment.offline_mode"
13
13
  JUDGMENT_UPDATE_ID = "judgment.update_id"
14
14
 
15
+ JUDGMENT_USAGE_METADATA = "judgment.usage.metadata"
16
+
15
17
  JUDGMENT_CUSTOMER_ID = "judgment.customer_id"
16
18
 
17
19
  JUDGMENT_AGENT_ID = "judgment.agent_id"
@@ -31,13 +33,15 @@ class AttributeKeys(str, Enum):
31
33
  GEN_AI_SYSTEM = "gen_ai.system"
32
34
  GEN_AI_USAGE_INPUT_TOKENS = "gen_ai.usage.input_tokens"
33
35
  GEN_AI_USAGE_OUTPUT_TOKENS = "gen_ai.usage.output_tokens"
34
- GEN_AI_USAGE_COMPLETION_TOKENS = "gen_ai.usage.completion_tokens"
36
+ GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS = (
37
+ "gen_ai.usage.cache_creation_input_tokens"
38
+ )
39
+ GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS = "gen_ai.usage.cache_read_input_tokens"
40
+
35
41
  GEN_AI_REQUEST_TEMPERATURE = "gen_ai.request.temperature"
36
42
  GEN_AI_REQUEST_MAX_TOKENS = "gen_ai.request.max_tokens"
37
43
  GEN_AI_RESPONSE_FINISH_REASONS = "gen_ai.response.finish_reasons"
38
44
 
39
- GEN_AI_USAGE_TOTAL_COST = "gen_ai.usage.total_cost_usd"
40
-
41
45
 
42
46
  class InternalAttributeKeys(str, Enum):
43
47
  """