From cbb78129b79ee6116613e9f086fb204a26928d89 Mon Sep 17 00:00:00 2001
From: Vinicius Mesel <4984147+vmesel@users.noreply.github.com>
Date: Wed, 30 Oct 2024 06:48:45 -0700
Subject: [PATCH] attempt 2: Tries to fix pipeline and fixes function return

---
 .github/workflows/pypi.yml | 9 ++++++---
 dialog_lib/db/memory.py    | 4 ++--
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/pypi.yml b/.github/workflows/pypi.yml
index e5e4ebb..601ca2b 100644
--- a/.github/workflows/pypi.yml
+++ b/.github/workflows/pypi.yml
@@ -19,6 +19,7 @@ jobs:
 
       - uses: actions/upload-artifact@v3
         with:
+          name: dist
           path: ./dist
 
   pypi-publish:
@@ -32,9 +33,11 @@ jobs:
       id-token: write
     steps:
       - uses: actions/download-artifact@v4.1.7
-      - name: Creates artifact directory
-        run: mkdir artifact
+        with:
+          name: dist
+          path: dist
+
       - name: Publish package distributions to PyPI
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
-          packages_dir: artifact/
\ No newline at end of file
+          packages_dir: dist/
\ No newline at end of file
diff --git a/dialog_lib/db/memory.py b/dialog_lib/db/memory.py
index 0d9282a..edd5043 100644
--- a/dialog_lib/db/memory.py
+++ b/dialog_lib/db/memory.py
@@ -107,7 +107,7 @@ def get_messages(self):
         get_messages_query = self._get_messages_query(self._table_name)
         for query in get_messages_query:
             self.cursor.execute(query)
-        return self.cursor.fetchall()
+        return [row[0] for row in self.cursor.fetchall()]
 
     async def aget_messages(self):
         """
@@ -118,7 +118,7 @@ async def aget_messages(self):
         async with async_conn.cursor() as cursor:
             for query in get_messages_query:
                 await cursor.execute(query)
-            return await cursor.fetchall()
+            return [row[0] for row in await cursor.fetchall()]
 
     def add_tags(self, tags: str) -> None:
         """