fda

babyagi/classesa/diamond.py

@@ -61,7 +61,62 @@ class ProductDatabase:
                 limit 10000000
             """, (query_vector, top_k))
             results = cursor.fetchall()
-            return results            
+            return results      
+
+def create_index():
+    # データベース接続情報
+    DATABASE_URL = "postgresql://miyataken999:[email protected]/neondb?sslmode=require"
+    
+    # ProductDatabaseクラスのインスタンスを作成
+    db = ProductDatabase(DATABASE_URL)
+    
+    # データベースに接続
+    db.connect()
+    
+    try:
+        # pgvector拡張機能のインストールとカラムの追加
+        db.setup_vector_extension_and_column()
+        print("Vector extension installed and column added successfully.")
+        query_text="1"
+        results = db.search_similar_all(query_text)
+        print("Search results:")
+        DEBUG=1
+        if DEBUG==1:
+            for result in results:
+                print(result) 
+                id = result[0]
+                sample_text = str(result[1])+str(result[2])+str(result[3])+str(result[4])+str(result[5])+str(result[6])+str(result[7])+str(result[8])+str(result[9])
+                print(sample_text)
+                db.insert_vector(id, sample_text) 
+        #return
+        # サンプルデータの挿入
+        #sample_text = """"""
+        #sample_product_id = 1  # 実際の製品IDを使用
+        #db.insert_vector(sample_product_id, sample_text)
+        #db.insert_vector(2, sample_text)
+
+        #print(f"Vector inserted for product ID {sample_product_id}.")
+
+        
+        # ベクトル検索
+        query_text = "2.03Very GoodJSI262.058.08.068.125.05"
+
+        query_text = "2.03Very GoodJSI2"
+        #query
+
+        #query_text = "2.03-Very Good-J-SI2-62.2-58.0-7.27-7.33-4.55"
+        results = db.search_similar_vectors(query)#query_text)
+        res_all = ""
+        print("Search results:")
+        for result in results:
+            print(result)
+            res_all += str(result)+"\r\n"
+        return res_all
+    
+    finally:
+        # 接続を閉じる
+        db.close()    
+
 
 def calculate(query):
     # データベース接続情報

mysite/routers/fastapi.py

@@ -121,6 +121,8 @@ def setup_webhook_routes(app: FastAPI):
                       ほかに、良い提案があればして下さい。こうしたらもっと、良くなるよなど。
                     ## リファスタの住所
                       〒170-0013 東京都豊島区東池袋１丁目２５−１４ アルファビルディング 4F
+                    ## ダイヤテンプレート
+                        - 
                     ## 買取テンプレート
                         ・ブランド名：
                         ・モデル名：