Sneed-Group/ai-brain
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update main.py

Browse source
This commit is contained in:
ThatLinuxFan 2024-01-03 10:37:19 -06:00 committed by GitHub
parent a4c26d93de
commit e7cb3d614f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
main.py
View file

@ -250,16 +250,16 @@ async def get_emote():
async def get_chatin(chatText: str): async def get_chatin(chatText: str):
return await main(chatText) return await main(chatText)
# Main function to run the program
# Main function to run the program # Main function to run the program
def main(chatin): def main(chatin):
chatin = "Guest:" + chatin chatin = "Guest:" + chatin
# Print ollama response and what it said
print(f"You: {chatin}")
print(f"AI:")
# Get response from ollama # Get response from ollama
message = llm(f"(Additional context for reply: {completeContext}), reply to this: {chatin}") message = llm(f"(Additional context for reply: {completeContext}), reply to this: {chatin}")
# Print ollama response and what it said
print(f"You: {chatin}")
print(f"AI: {message}")
return f"You: {chatin}\n${name}: ${message}" return f"You: {chatin}\n${name}: ${message}"