Revert "fix: test sdk bump"

This reverts commit ed0fa66.
epam · Mar 5, 2024 · 508357c · 508357c
1 parent ed0fa66
commit 508357c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/aidial_adapter_vertexai/chat_completion.py b/aidial_adapter_vertexai/chat_completion.py
@@ -84,4 +84,4 @@ async def generate_response(usage: TokenUsage, choice_idx: int) -> None:
         response.set_usage(usage.prompt_tokens, usage.completion_tokens)
 
         if params.max_prompt_tokens is not None:
-            response.set_discarded_messages([])
+            response.set_discarded_messages(discarded_messages_count)
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,7 +15,7 @@ repository = "https://github.com/epam/ai-dial-adapter-vertexai/"
 
 [tool.poetry.dependencies]
 python = "~3.11"
-aidial-sdk = {version = "0.7.0", extras = ["telemetry"]}
+aidial-sdk = {version = "0.6.2", extras = ["telemetry"]}
 fastapi = "0.109.2"
 google-cloud-aiplatform = "1.38.1"
 google-auth = "2.21.0"