From 73a476979e96b1475d409dfc7542fe0603856e44 Mon Sep 17 00:00:00 2001
From: mst <137482396+osushinekotan@users.noreply.github.com>
Date: Wed, 11 Sep 2024 22:47:03 +0900
Subject: [PATCH] fix: change column type to string for relation_type (#272)
 #none

---
 libs/ktem/ktem/index/file/index.py   | 2 +-
 scripts/migrate/migrate_chroma_db.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/libs/ktem/ktem/index/file/index.py b/libs/ktem/ktem/index/file/index.py
index 2f1bc8a..0cbd57f 100644
--- a/libs/ktem/ktem/index/file/index.py
+++ b/libs/ktem/ktem/index/file/index.py
@@ -115,7 +115,7 @@ class FileIndex(BaseIndex):
                 "id": Column(Integer, primary_key=True, autoincrement=True),
                 "source_id": Column(String),
                 "target_id": Column(String),
-                "relation_type": Column(Integer),
+                "relation_type": Column(String),
                 "user": Column(Integer, default=1),
             },
         )
diff --git a/scripts/migrate/migrate_chroma_db.py b/scripts/migrate/migrate_chroma_db.py
index 96fa920..adffca0 100644
--- a/scripts/migrate/migrate_chroma_db.py
+++ b/scripts/migrate/migrate_chroma_db.py
@@ -83,7 +83,7 @@ def _init_resource(private: bool = True, id: int = 1):
             "id": Column(Integer, primary_key=True, autoincrement=True),
             "source_id": Column(String),
             "target_id": Column(String),
-            "relation_type": Column(Integer),
+            "relation_type": Column(String),
             "user": Column(Integer, default=1),
         },
     )