Skip to content

Commit 4078571

Browse files
committed
Update header
1 parent 52401d3 commit 4078571

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

text_2_sql/text_2_sql_core/src/text_2_sql_core/payloads/agent_request_response_pair.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from datetime import datetime, timezone
88

99

10-
class AgentResponseHeader(BaseModel):
10+
class AgentRequestResponseHeader(BaseModel):
1111
prompt_tokens: int
1212
completion_tokens: int
1313
timestamp: datetime = Field(
@@ -84,6 +84,6 @@ def add_defaults_to_injected_parameters(cls, values):
8484

8585

8686
class AgentRequestResponsePair(BaseModel):
87-
header: AgentResponseHeader | None = Field(default=None)
87+
header: AgentRequestResponseHeader | None = Field(default=None)
8888
request: AgentRequestBody
8989
response: AgentResponseBody

0 commit comments

Comments
 (0)