AURA GenAI Libs

Introduction

aura-genai-libs is a ATRIA that contains common components for agent development, such as logs and the langchain call to the atria_model_gw service.

Includes the following modules:

• logger module, whose purpose is to implement a callback handler to log the different stages of model, chain, and tool processing within the LangChain framework, integrating it with the AURA PyTraces logging system and the ATRIA Agents context. The module is based on the use of BaseCallbackHandler (langchain_core) and logs information at different stages of the execution cycle of language models (LLM), chats, chains, and tools. Each logged event uses m_headers.

• chat_models module, which provides a custom implementation of the ChatOpenAI class from LangChain. This custom implementation is designed to facilitate the integration with the atria_model_gw service.

How to use it

To use the aura-genai-libs, you need to install it in your Python environment and import and use the module in your code.