React-Group/ai-virtual-assistant
Archived
3
2
Fork 2
Code Issues 24 Pull requests Projects Releases Packages Wiki Activity Actions

Expanded AI API.

Browse source
This commit is contained in:
Patrick_Pluto 2024-09-16 11:27:00 +02:00
parent 7cdcc35a2a
commit c4655fb49e
7 changed files with 52 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
py/.idea/.gitignore generated vendored Normal file
View file

@ -0,0 +1,3 @@
# Default ignored files
/shelf/
/workspace.xml

6
py/.idea/inspectionProfiles/profiles_settings.xml generated Normal file
View file

@ -0,0 +1,6 @@
<component name="InspectionProjectProfileManager">
<settings>
<option name="USE_PROJECT_PROFILE" value="false" />
<version value="1.0" />
</settings>
</component>

7
py/.idea/misc.xml generated Normal file
View file

@ -0,0 +1,7 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="Black">
<option name="sdkName" value="Python 3.12" />
</component>
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.12" project-jdk-type="Python SDK" />
</project>

8
py/.idea/modules.xml generated Normal file
View file

@ -0,0 +1,8 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="ProjectModuleManager">
<modules>
<module fileurl="file://$PROJECT_DIR$/.idea/py.iml" filepath="$PROJECT_DIR$/.idea/py.iml" />
</modules>
</component>
</project>

8
py/.idea/py.iml generated Normal file
View file

@ -0,0 +1,8 @@
<?xml version="1.0" encoding="UTF-8"?>
<module type="PYTHON_MODULE" version="4">
<component name="NewModuleRootManager">
<content url="file://$MODULE_DIR$" />
<orderEntry type="inheritedJdk" />
<orderEntry type="sourceFolder" forTests="false" />
</component>
</module>

6
py/.idea/vcs.xml generated Normal file
View file

@ -0,0 +1,6 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="VcsDirectoryMappings">
<mapping directory="$PROJECT_DIR$/.." vcs="Git" />
</component>
</project>

19
py/api.py
View file

@ -2,10 +2,10 @@ import requests
import json
class api:
def initialize_ollama(self, prompt):
class API:
@staticmethod
def process_text(prompt, model):
ollama_url = "http://localhost:11434"
model = "phi3.5"
response = requests.post(
f"{ollama_url}/api/generate", json={"model": model, "prompt": prompt}
@ -24,6 +24,15 @@ class api:
else:
return "Error: " + response.text
def send_message(self, message):
answer = self.initialize_ollama(message)
def send_message(self, message, model):
if model == 1:
answer = self.process_text(message, "phi3.5")
elif model == 2:
answer = self.process_text(message, "gemma2:2b")
elif model == 3:
answer = self.process_text(message, "qwen2:0.5b")
elif model == 4:
answer = self.process_text(message, "codegemma:2b")
else:
return "Invalid choice"
return answer