pip3 install lytix-py

npm install @lytix/client

from optimodel import LytixCreds

LytixCreds.setAPIKey('<your-api-key-here>')

import { LytixCred } from "@lytix/client";

LytixCreds.setAPIKey("sk-1234");

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())

import { MetricCollector } from "@lytix/client";

await MetricCollector.captureModelIO({
  modelName: "testModelName",
  systemPrompt: "You area helpful assistant",
  userPrompt: "Some user input",
  assistantMessage: "Some model output",


  /**
   * Optional fields
   */
  metricMetadata: {
    /**
     * Optional extra data
     */
    env: "prod",
  }
  userIdentifier?: string;
  sessionId?: string;
);