diff --git a/add_user.py b/add_user.py index be7c364..6a600fd 100644 --- a/add_user.py +++ b/add_user.py @@ -7,10 +7,16 @@ import asyncio from tortoise import Tortoise from blueprints.users.models import User +from dotenv import load_dotenv + +load_dotenv() + # Database configuration with environment variable support DATABASE_PATH = os.getenv("DATABASE_PATH", "database/raggr.db") DATABASE_URL = os.getenv("DATABASE_URL", f"sqlite://{DATABASE_PATH}") +print(DATABASE_URL) + async def add_user(username: str, email: str, password: str): """Add a new user to the database""" diff --git a/main.py b/main.py index 6c88ded..5ee4740 100644 --- a/main.py +++ b/main.py @@ -186,7 +186,7 @@ def consult_oracle( def llm_chat(input: str, transcript: str = "") -> str: system_prompt = "You are a helpful assistant that understands veterinary terms." transcript_prompt = f"Here is the message transcript thus far {transcript}." - prompt = f"""Answer the user in a humorous way as if you were a cat named Simba. Be very coy. + prompt = f"""Answer the user in as if you were a cat named Simba. Don't act too catlike. Be assertive. {transcript_prompt if len(transcript) > 0 else ""} Respond to this prompt: {input}""" output = llm_client.chat(prompt=prompt, system_prompt=system_prompt)