OpenAI Integration

import openai
import requests

# AIVory Guard API configuration
AIVORY_BASE_URL = "https://app.aivory.net"
AIVORY_API_KEY = "your_aivory_api_key"

def scan_code_with_aivory(code, language, filename="main.py"):
    """Scan code using AIVory Guard API"""
    headers = {
        "Authorization": f"Bearer {AIVORY_API_KEY}",
        "Content-Type": "application/json"
    }

    payload = {
        "code": code,
        "language": language,
        "filename": filename,
        "standards": ["OWASP", "GDPR", "HIPAA"]
    }

    response = requests.post(
        f"{AIVORY_BASE_URL}/api/scan",
        headers=headers,
        json=payload
    )

    return response.json()

# Use in OpenAI Assistant
client = openai.OpenAI(api_key="your_openai_key")

# Create an assistant with code scanning capability
assistant = client.beta.assistants.create(
    name="Secure Code Generator",
    instructions="""You are a secure code generator. When writing code:
    1. Generate the code as requested
    2. Scan it using the scan_code_with_aivory function
    3. Review the results and fix any violations found
    4. Re-scan to verify fixes before completing the task

    Always report both violations found and fixes applied.""",
    model="gpt-4",
    tools=[{"type": "function", "function": {
        "name": "scan_code_with_aivory",
        "description": "Scan generated code for security and compliance violations",
        "parameters": {
            "type": "object",
            "properties": {
                "code": {"type": "string", "description": "The code to scan"},
                "language": {"type": "string", "description": "Programming language"},
                "filename": {"type": "string", "description": "Filename"}
            },
            "required": ["code", "language"]
        }
    }}]
)

import OpenAI from 'openai';
import fetch from 'node-fetch';

const AIVORY_BASE_URL = 'https://app.aivory.net';
const AIVORY_API_KEY = process.env.AIVORY_API_KEY;

async function scanCodeWithAIVory(code: string, language: string, filename: string = 'main.js') {
  const response = await fetch(`${AIVORY_BASE_URL}/api/scan`, {
    method: 'POST',
    headers: {
      'Authorization': `Bearer ${AIVORY_API_KEY}`,
      'Content-Type': 'application/json'
    },
    body: JSON.stringify({
      code,
      language,
      filename,
      standards: ['OWASP', 'GDPR', 'PCI-DSS']
    })
  });

  return await response.json();
}

// Use with OpenAI
const openai = new OpenAI({
  apiKey: process.env.OPENAI_API_KEY
});

const assistant = await openai.beta.assistants.create({
  name: 'Secure Code Generator',
  instructions: 'Generate secure code and scan it for compliance violations using the scan_code function.',
  model: 'gpt-4',
  tools: [{
    type: 'function',
    function: {
      name: 'scan_code',
      description: 'Scan code for security and compliance violations',
      parameters: {
        type: 'object',
        properties: {
          code: { type: 'string' },
          language: { type: 'string' },
          filename: { type: 'string' }
        },
        required: ['code', 'language']
      }
    }
  }]
});

from openai_agent import Agent

class SecureCodeAgent(Agent):
    def __init__(self):
        super().__init__(name="Secure Code Agent")
        self.aivory_api_key = os.getenv("AIVORY_API_KEY")

    def generate_and_scan_code(self, prompt):
        # Generate code with OpenAI
        code = self.generate_code(prompt)

        # Scan with AIVory Guard
        scan_result = self.scan_code(code)

        # Return results
        return {
            "code": code,
            "scan_result": scan_result
        }

    def scan_code(self, code, language="python"):
        import requests
        response = requests.post(
            "https://app.aivory.net/api/scan",
            headers={"Authorization": f"Bearer {self.aivory_api_key}"},
            json={"code": code, "language": language}
        )
        return response.json()

def scan_code_safe(code, language):
    try:
        response = requests.post(
            f"{AIVORY_BASE_URL}/api/scan",
            headers={"Authorization": f"Bearer {AIVORY_API_KEY}"},
            json={"code": code, "language": language},
            timeout=30
        )
        response.raise_for_status()
        return response.json()
    except requests.exceptions.Timeout:
        return {"error": "Scan timeout - code may be too large"}
    except requests.exceptions.HTTPError as e:
        return {"error": f"API error: {e.response.status_code}"}
    except Exception as e:
        return {"error": f"Unexpected error: {str(e)}"}

import logging
logging.basicConfig(level=logging.DEBUG)

# Your API calls will now show detailed request/response info