Merge pull request #72 from dlimeng/dev

model python web_utils

Author: dlimeng

solidui-dao/src/main/java/com/cloudorc/solidui/dao/entity/DataSourceTypeKey.java

@@ -79,6 +79,9 @@ public enum Scope {
     @TableId(value = "id", type = IdType.AUTO)
     private Long id;
 
+    @TableId(value = "datasource_type_id")
+    private Long dataSourceTypeId;
+
     /** Key name */
     private String key;
 
@@ -92,7 +95,6 @@ public enum Scope {
 
     private String nameEn;
 
-
     /** Default value */
     private String defaultValue;
 
@@ -105,8 +107,6 @@ public enum Scope {
     /** Value regex */
     private String valueRegex;
 
-
-
     public Long getId() {
         return id;
     }
@@ -195,6 +195,14 @@ public void setNameEn(String nameEn) {
         this.nameEn = nameEn;
     }
 
+    public Long getDataSourceTypeId() {
+        return dataSourceTypeId;
+    }
+
+    public void setDataSourceTypeId(Long dataSourceTypeId) {
+        this.dataSourceTypeId = dataSourceTypeId;
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) {

soliduimodelui/kernelprogram/main.py

@@ -60,7 +60,7 @@
 cli.show_server_banner = lambda *x: None
 
 app = Flask(__name__)
-app.register_blueprint(base_blueprint)
+
 CORS(app)
 
 def start_kernel_manager():
@@ -153,6 +153,8 @@ def handle_restart():
     return jsonify({"result": "success"})
 
 
+app.register_blueprint(base_blueprint)
+
 async def main():
     start_kernel_manager()
 
@@ -170,6 +172,3 @@ def run_flask_app():
 if __name__ == "__main__":
     asyncio.run(main())
 
-
-
-    

soliduimodelui/kernelprogram/utils.py

@@ -40,4 +40,5 @@ def init_snakemq(ident, init_type="listen"):
         link.add_connector(("localhost", config.SNAKEMQ_PORT))
     else:
         raise Exception("Unsupported init type.")
-    return messaging, link
+    return messaging, link
+

soliduimodelui/webapp/main.py

@@ -26,7 +26,10 @@
 from flask_cors import CORS
 from flask import Flask, request, jsonify, send_from_directory, Response, Blueprint
 from dotenv import load_dotenv
-
+import soliduimodelui.webapp.web_utils as web_utils
+# from langchain.prompts import PromptTemplate
+# from langchain.llms import OpenAI
+# from langchain.chains import LLMChain
 
 load_dotenv('.env')
 
@@ -42,10 +45,21 @@
 cli.show_server_banner = lambda *x: None
 
 app = Flask(__name__)
-app.register_blueprint(base_blueprint)
 CORS(app)
 
 
+async def get_code(user_prompt, user_key=None, model="gpt-3.5-turbo", type=0):
+    return 0
+
+@base_blueprint.route('/api/<path:path>', methods=["GET", "POST"])
+def proxy_kernel_manager(path):
+    return web_utils.response_format()
+
+@base_blueprint.route('/generate', methods=['POST'])
+def generate_code():
+    return web_utils.response_format()
+
+app.register_blueprint(base_blueprint)
 
 if __name__ == '__main__':
     app.run(host="0.0.0.0", port=APP_PORT, debug=True, use_reloader=False)

soliduimodelui/webapp/web_utils.py

@@ -0,0 +1,22 @@
+#
+# Licensed to the Apache Software Foundation (ASF) under one or more
+# contributor license agreements.  See the NOTICE file distributed with
+# this work for additional information regarding copyright ownership.
+# The ASF licenses this file to You under the Apache License, Version 2.0
+# (the "License"); you may not use this file except in compliance with
+# the License.  You may obtain a copy of the License at
+# http://www.apache.org/licenses/LICENSE-2.0
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+def response_format(code=0,msg="success",data={},success=True,failed=False):
+    return {
+        "code": code,
+        "msg": msg,
+        "data": data,
+        "success": success,
+        "failed": failed
+    }