From 802064aee72b03ab38ead0cda780cfa3e37ce728 Mon Sep 17 00:00:00 2001
From: Bryce Freshcorn <26725654+brycecf@users.noreply.github.com>
Date: Thu, 25 Apr 2024 00:36:09 -0400
Subject: [PATCH] Corrected wrong output type for `OutputKeys.from_keys()`
 (#13086)

* Corrected wrong output type for `OutputKeys.from_keys`. Incremented `core` version to `0.10.32`.

* Revert "Corrected wrong output type for `OutputKeys.from_keys`. Incremented `core` version to `0.10.32`."

This reverts commit 1e6391638053ea460f9cc01a2a424efa2a029a75.

* Corrected wrong output type for `OutputKeys.from_keys`.

* Update query.py

---------

Co-authored-by: Andrei Fajardo <92402603+nerdai@users.noreply.github.com>
---
 .../llama_index/core/base/query_pipeline/query.py             | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/llama-index-core/llama_index/core/base/query_pipeline/query.py b/llama-index-core/llama_index/core/base/query_pipeline/query.py
index a26bf69691f73..4300d9a06b63a 100644
--- a/llama-index-core/llama_index/core/base/query_pipeline/query.py
+++ b/llama-index-core/llama_index/core/base/query_pipeline/query.py
@@ -107,8 +107,8 @@ class OutputKeys(BaseModel):
     def from_keys(
         cls,
         required_keys: Set[str],
-    ) -> "InputKeys":
-        """Create InputKeys from tuple."""
+    ) -> "OutputKeys":
+        """Create OutputKeys from tuple."""
         return cls(required_keys=required_keys)
 
     def validate(self, input_keys: Set[str]) -> None: