import os
import base64
from pathlib import Path
from dotenv import load_dotenv
load_dotenv()
XAI_API_KEY = os.getenv("XAI_API_KEY", "")
if XAI_API_KEY:
os.environ["XAI_API_KEY"] = XAI_API_KEY
def encode_image_to_base64(image_path: str) -> str:
path = Path(image_path)
if not path.exists():
raise FileNotFoundError(f"Image file not found: {image_path}")
with open(image_path, "rb") as image_file:
return base64.b64encode(image_file.read()).decode("utf-8")
def extract_usage(response):
if not response.usage:
return {}
return {
"prompt_tokens": response.usage.prompt_tokens,
"completion_tokens": response.usage.completion_tokens,
"reasoning_tokens": response.usage.reasoning_tokens,
"total_tokens": response.usage.total_tokens,
}
def build_params(**kwargs):
result = {}
for key, value in kwargs.items():
if value:
result[key] = value
return result
