feat:dockerfile添加以及向量数据库支持配置为远程数据库

main_service.py

@@ -38,9 +38,11 @@ def load_train_data_ddl(vn: CustomVanna):
 
 
 def create_vana():
-    print("----------------create---------")
+    logger.info("----------------create vana ---------")
+    q_client = QdrantClient(":memory:") if config('QDRANT_TYPE', default='memory') == 'memory' else QdrantClient(
+        url=config('QDRANT_DB_HOST', default=''), port=config('QDRANT_DB_PORT', default=6333))
     vn = CustomVanna(
-        vector_store_config={"client": QdrantClient(":memory:")},
+        vector_store_config={"client": q_client},
         llm_config={
             "api_key": config('CHAT_MODEL_API_KEY', default=''),
             "api_base": config('CHAT_MODEL_BASE_URL', default=''),
@@ -51,7 +53,7 @@ def create_vana():
 
 
 def init_vn(vn):
-    print("--------------init vn-----connect----")
+    logger.info("--------------init vana-----connect to datasouce db----")
     connect_database(vn)
     load_ddl_doc.add_ddl(vn)
     load_ddl_doc.add_documentation(vn)
@@ -100,7 +102,7 @@ def generate_sql_2():
         data = vn.generate_sql_2(question=question)
         data['id'] = id
         sql = data["resp"]["sql"]
-        print("sql:", sql)
+        logger.info("generate sql is : "+ sql)
         cache.set(id=id, field="question", value=question)
         cache.set(id=id, field="sql", value=sql)
         print("data---------------------------", data)