diff --git a/llama-index-integrations/llms/llama-index-llms-openai/llama_index/llms/openai/base.py b/llama-index-integrations/llms/llama-index-llms-openai/llama_index/llms/openai/base.py index ae0bcae46af9d25d16bdf849011d06c9982db897..b62747c0f1070500042839b94350b2c816e36074 100644 --- a/llama-index-integrations/llms/llama-index-llms-openai/llama_index/llms/openai/base.py +++ b/llama-index-integrations/llms/llama-index-llms-openai/llama_index/llms/openai/base.py @@ -666,10 +666,17 @@ class OpenAI(LLM): stream=False, **all_kwargs, ) + text = response.choices[0].text + openai_completion_logprobs = response.choices[0].logprobs + logprobs = None + if openai_completion_logprobs: + logprobs = from_openai_completion_logprobs(openai_completion_logprobs) + return CompletionResponse( text=text, raw=response, + logprobs=logprobs, additional_kwargs=self._get_response_token_counts(response), ) diff --git a/llama-index-integrations/llms/llama-index-llms-openai/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-openai/pyproject.toml index 010b96ae105c620385a9c4ed50a44d1fc0f36e46..b9715befa66fb331f36f54aed093347c4f30e702 100644 --- a/llama-index-integrations/llms/llama-index-llms-openai/pyproject.toml +++ b/llama-index-integrations/llms/llama-index-llms-openai/pyproject.toml @@ -29,11 +29,11 @@ exclude = ["**/BUILD"] license = "MIT" name = "llama-index-llms-openai" readme = "README.md" -version = "0.1.11" +version = "0.1.12" [tool.poetry.dependencies] python = ">=3.8.1,<4.0" -llama-index-core = "^0.10.19" +llama-index-core = "^0.10.20.post1" [tool.poetry.group.dev.dependencies] ipython = "8.10.0"