From a77da4192c24074fa8417be1ebb3b1006f546c07 Mon Sep 17 00:00:00 2001
From: Andrei Fajardo <92402603+nerdai@users.noreply.github.com>
Date: Tue, 12 Mar 2024 13:30:53 -0400
Subject: [PATCH] Add logprobs field to `CompletionResponse` (#11855)
---
llama-index-core/llama_index/core/base/llms/types.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/llama-index-core/llama_index/core/base/llms/types.py b/llama-index-core/llama_index/core/base/llms/types.py
index 012c3c04bd..3ee9adf085 100644
--- a/llama-index-core/llama_index/core/base/llms/types.py
+++ b/llama-index-core/llama_index/core/base/llms/types.py
@@ -83,6 +83,7 @@ class CompletionResponse(BaseModel):
text: str
additional_kwargs: dict = Field(default_factory=dict)
raw: Optional[dict] = None
+ logprobs: Optional[List[List[LogProb]]] = None
delta: Optional[str] = None
def __str__(self) -> str:
--
GitLab