Serverless Deployment

FROM ghcr.io/tealtiger/python-sdk:latest

# Copy your application code
COPY app.py ${LAMBDA_TASK_ROOT}/

# Set the CMD to your handler
CMD ["app.handler"]

# app.py
from tealtiger import TealOpenAI, PolicyMode

def handler(event, context):
    client = TealOpenAI(
        api_key=os.environ['OPENAI_API_KEY'],
        mode=PolicyMode.ENFORCE
    )
    
    response = client.chat.completions.create(
        model="gpt-4",
        messages=[{"role": "user", "content": event['prompt']}]
    )
    
    return {
        'statusCode': 200,
        'body': response.choices[0].message.content
    }

# Build and push image
docker build -t tealtiger-lambda .
docker tag tealtiger-lambda:latest 123456789012.dkr.ecr.us-east-1.amazonaws.com/tealtiger-lambda:latest
docker push 123456789012.dkr.ecr.us-east-1.amazonaws.com/tealtiger-lambda:latest

# Create Lambda function
aws lambda create-function \
  --function-name tealtiger-app \
  --package-type Image \
  --code ImageUri=123456789012.dkr.ecr.us-east-1.amazonaws.com/tealtiger-lambda:latest \
  --role arn:aws:iam::123456789012:role/lambda-execution-role \
  --timeout 60 \
  --memory-size 512 \
  --environment Variables={OPENAI_API_KEY=sk-xxx}

# Get layer ARN for Python 3.11 in us-east-1
curl https://layers.tealtiger.ai/api/layers/python3.11/us-east-1

# Response:
{
  "runtime": "python3.11",
  "region": "us-east-1",
  "version": "1.1.0",
  "arn": "arn:aws:lambda:us-east-1:123456789012:layer:tealtiger-python311:1"
}

# app.py (no dependencies needed!)
from tealtiger import TealOpenAI

def handler(event, context):
    client = TealOpenAI(api_key=os.environ['OPENAI_API_KEY'])
    # Your code here

# Create deployment package (just your code)
zip function.zip app.py

# Create function with layer
aws lambda create-function \
  --function-name tealtiger-app \
  --runtime python3.11 \
  --handler app.handler \
  --zip-file fileb://function.zip \
  --role arn:aws:iam::123456789012:role/lambda-execution-role \
  --layers arn:aws:lambda:us-east-1:123456789012:layer:tealtiger-python311:1 \
  --environment Variables={OPENAI_API_KEY=sk-xxx}

# main.tf
resource "aws_lambda_function" "tealtiger_app" {
  function_name = "tealtiger-app"
  role          = aws_iam_role.lambda_role.arn
  
  # Option 1: Container image
  package_type = "Image"
  image_uri    = "123456789012.dkr.ecr.us-east-1.amazonaws.com/tealtiger-lambda:latest"
  
  # Option 2: Lambda layer
  # filename      = "function.zip"
  # runtime       = "python3.11"
  # handler       = "app.handler"
  # layers        = [data.aws_lambda_layer_version.tealtiger.arn]
  
  timeout     = 60
  memory_size = 512
  
  environment {
    variables = {
      OPENAI_API_KEY = var.openai_api_key
    }
  }
}

# Get TealTiger layer ARN
data "aws_lambda_layer_version" "tealtiger" {
  layer_name = "tealtiger-python311"
}