diff --git a/.gitignore b/.gitignore index e28c376..1de8943 100644 --- a/.gitignore +++ b/.gitignore @@ -35,6 +35,4 @@ yarn-error.log* *.tsbuildinfo next-env.d.ts - -.idea/ -venv/ +api_key.txt \ No newline at end of file diff --git a/py/.idea/.gitignore b/py/.idea/.gitignore deleted file mode 100644 index 26d3352..0000000 --- a/py/.idea/.gitignore +++ /dev/null @@ -1,3 +0,0 @@ -# Default ignored files -/shelf/ -/workspace.xml diff --git a/py/.idea/inspectionProfiles/profiles_settings.xml b/py/.idea/inspectionProfiles/profiles_settings.xml deleted file mode 100644 index 105ce2d..0000000 --- a/py/.idea/inspectionProfiles/profiles_settings.xml +++ /dev/null @@ -1,6 +0,0 @@ - - - - \ No newline at end of file diff --git a/py/.idea/misc.xml b/py/.idea/misc.xml deleted file mode 100644 index 3671ece..0000000 --- a/py/.idea/misc.xml +++ /dev/null @@ -1,10 +0,0 @@ - - - - - - - - \ No newline at end of file diff --git a/py/.idea/modules.xml b/py/.idea/modules.xml deleted file mode 100644 index 3a65488..0000000 --- a/py/.idea/modules.xml +++ /dev/null @@ -1,8 +0,0 @@ - - - - - - - - \ No newline at end of file diff --git a/py/.idea/py.iml b/py/.idea/py.iml deleted file mode 100644 index 49f4c24..0000000 --- a/py/.idea/py.iml +++ /dev/null @@ -1,13 +0,0 @@ - - - - - - - - - - - - \ No newline at end of file diff --git a/py/.idea/vcs.xml b/py/.idea/vcs.xml deleted file mode 100644 index 6c0b863..0000000 --- a/py/.idea/vcs.xml +++ /dev/null @@ -1,6 +0,0 @@ - - - - - - \ No newline at end of file diff --git a/py/api.py b/py/api.py deleted file mode 100644 index 6eb7cc0..0000000 --- a/py/api.py +++ /dev/null @@ -1,51 +0,0 @@ -from flask import Flask, request, jsonify -import ollama - - -class AI: - @staticmethod - def process_local(model, message, system, return_class, access_token): - stream = ollama.chat( - model=model, - messages=[{'role': 'user', 'content': message}, {'role': 'system', 'content': system}], - stream=True, - ) - - for chunk in stream: - print(chunk['message']['content']) - return_class.ai_response[access_token] += chunk['message']['content'] - - -class API: - def __init__(self): - self.app = Flask(__name__) - self.ai_response = [] - self.ai = AI() - - def run(self): - @self.app.route('/interstellar/api/ai_create', methods=['GET']) - def create_ai(): - self.ai_response.append("") - return jsonify({'status': 200, 'access_token': len(self.ai_response) - 1}) - - @self.app.route('/interstellar/api/ai_send', methods=['POST']) - def send_ai(): - data = request.get_json() - message = data.get('message') - ai_model = data.get('ai_model') - system_prompt = data.get('system_prompt') - access_token = data.get('access_token') - self.ai.process_local(ai_model, message, system_prompt, self, access_token) - return jsonify({'status': 200}) - - @self.app.route('/interstellar/api/ai_get', methods=['GET']) - def get_ai(): - data = request.args.get('access_token') - return jsonify({'status': 200, 'response': self.ai_response[int(data)]}) - - self.app.run(debug=True) - - -if __name__ == '__main__': - api = API() - api.run() diff --git a/py/install.sh b/py/install.sh deleted file mode 100644 index 1fbdcba..0000000 --- a/py/install.sh +++ /dev/null @@ -1,4 +0,0 @@ -python -m venv venv -source venv/bin/activate -pip install -r requirements.txt -deactivate \ No newline at end of file diff --git a/py/requirements.txt b/py/requirements.txt deleted file mode 100644 index 731af42..0000000 --- a/py/requirements.txt +++ /dev/null @@ -1,2 +0,0 @@ -flask -ollama \ No newline at end of file