AI Llama Module

ai_llama_module is a lightweight Python wrapper for the Llama AI API, allowing you to easily send messages and receive AI-generated responses.

⸻

Installation

Install via pip:

pip install ai_llama_module

Requirements: • Python 3.9+ • requests • python-dotenv

Make sure you have an environment variable llama_ai set with your Llama AI API key, e.g., in a .env file:

You can also pass an API key through with the variable api_key when calling the function, otherwise it will default to the enviornment variable.

llama_ai = your_api_key_here

⸻

Usage

from ai_llama_module import call_ai

Single message

response = call_ai("Hello, Llama!")

print(response)

or you can use

response = call_ai("Hello World!", model, api_key = "YOUR KEY HERE")

print(response)

Multiple messages

messages = [
    {"role": "user", "content": "Hello!"},
    {"role": "assistant", "content": "Hi there! How can I help?"}
]

response = call_ai(messages)

print(response)

⸻

Features:

•	Supports single and multi-message inputs
•	Automatically reads API key from environment variables
•	Simple and easy-to-use interface
•	You can specify the model used with the variable 'model' which defualts to Llama-4-Maverick-17B-128E-Instruct-FP8

⸻

Dependencies

•	requests
•	python-dotenv

⸻

License

MIT License – see LICENSE for details.