aryn-ai · karanataryn · Oct 14, 2024 · Oct 12, 2024
diff --git a/lib/sycamore/poetry.lock b/lib/sycamore/poetry.lock
diff --git a/lib/sycamore/pyproject.toml b/lib/sycamore/pyproject.toml
@@ -62,7 +62,7 @@ pypdf = "^4.3.0"
 
 # Connector dependencies
 elasticsearch = { version = "8.14.0", optional = true}
-duckdb = { version = "^1.0.0", optional = true }
+duckdb = { version = "^1.1.1", optional = true }
 neo4j = { version = "^5.21.0", optional = true }
 opensearch-py = { version = "^2.3.1", optional = true }
 pinecone-client = { version = "^4.1.0", extras = ["grpc"], optional = true }

diff --git a/lib/sycamore/sycamore/connectors/duckdb/duckdb_reader.py b/lib/sycamore/sycamore/connectors/duckdb/duckdb_reader.py
@@ -64,13 +64,13 @@ class DuckDBReaderQueryResponse(BaseDBReader.QueryResponse):
 
     def to_docs(self, query_params: "BaseDBReader.QueryParams") -> list[Document]:
         assert isinstance(self, DuckDBReaderQueryResponse)
-        data = self.output.fetchdf()
+        data = self.output.df()
         data = data.to_dict(orient="records")
         result = []
         for object in data:
             val = object.get("properties")
             if val is not None:
-                object["properties"] = convert_from_str_dict(dict(zip(val["key"], val["value"])))
+                object["properties"] = convert_from_str_dict(val)
             result.append(Document(object))
         return result