import asyncio
from lytix_py import lytix
userInput = "Whats the capital of France?"
"""
The function to be traced
"""
@lytix.trace(modelName="testModelName")
async def getResponse(logger):
modelOutput = "Paris is the capital of france"
# Set our defined params however we want
lytix.setMessage(userInput, "user")
# Set a system message
lytix.setMessage("You are a helpful assistant", "system")
# Set the output
lytix.setMessage("Ouput of model", "assistant")
# Optional values
lytix.setUserIdentifier("testUser")
lytix.setSessionId("testSession")
return modelOutput
async def main():
response = await getResponse("What is the capital of France?")
print(response)
if __name__ == "__main__":
asyncio.run(main())
