edg4llm 1.0.4

A unified tool to generate fine-tuning datasets for LLMs, including questions, answers, and dialogues.

EDG4LLM

Table of Contents

• Features
• Installation
• Library Scope
• Quick Start
• Requirements
• Development Setup
• Testing
• License
• Security
• Contributing

Features

• Convert OpenAPI specifications into LLM-compatible tool/function definitions
• Support for multiple LLM providers (OpenAI, Anthropic, Cohere)
• Handle complex request bodies and parameter types
• Support for multiple authentication mechanisms
• Support for OpenAPI 3.0.x and 3.1.x specifications
• Handles both YAML and JSON OpenAPI specifications

Installation

pip install openapi-llm

Supported Python Versions

• Python >= 3.8

LLM Provider Dependencies

This library focuses on OpenAPI-to-LLM conversion and doesn't include LLM provider libraries by default. Install the ones you need:

# For OpenAI
pip install openai

# For Anthropic
pip install anthropic

# For Cohere
pip install cohere

Library Scope

OpenAPI-LLM provides core functionality for converting OpenAPI specifications into LLM-compatible tool/function definitions. It intentionally does not provide an opinionated, high-level interface for OpenAPI-LLM interactions. Users are encouraged to develop their own thin application layer above this library that suits their specific needs and preferences for OpenAPI-LLM integration.

OpenAPI Specification Validation

This library does not perform OpenAPI specification validation. It is the user's responsibility to ensure that the provided OpenAPI specifications are valid. We recommend using established validation tools such as:

• openapi-spec-validator
• prance
• Swagger Editor

Example of validating a spec before using it with openapi-llm:

from openapi_spec_validator import validate_spec
import yaml

# Load and validate your OpenAPI spec
with open('your_spec.yaml', 'r') as f:
    spec_dict = yaml.safe_load(f)
validate_spec(spec_dict)

Quick Start

Here's a practical example using OpenAI to perform a Google search via SerperDev API:

import os
from openai import OpenAI

from openapi_llm.client.config import ClientConfig
from openapi_llm.client.openapi import OpenAPIClient
from openapi_llm.core.spec import OpenAPISpecification


# Configure the OpenAPI client with SerperDev API spec and credentials
config = ClientConfig(
    openapi_spec=OpenAPISpecification.from_url("https://bit.ly/serperdev_openapi"), 
    credentials=os.getenv("SERPERDEV_API_KEY")
)

# Initialize OpenAI client
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

# Create a chat completion with tool definitions
response = client.chat.completions.create(
    model="gpt-4o",
    messages=[{"role": "user", "content": "Do a serperdev google search: Who was Nikola Tesla?"}],
    tools=config.get_tool_definitions(),
)

# Execute the API call based on the LLM's response
service_api = OpenAPIClient(config)
service_response = service_api.invoke(response)

This example demonstrates:

• Loading an OpenAPI specification from a URL
• Integrating with OpenAI's function calling
• Handling API authentication
• Converting and executing OpenAPI calls based on LLM responses

Requirements

• Python >= 3.8
• Dependencies:
  • jsonref
  • requests
  • PyYAML

Development Setup

1. Clone the repository

git clone https://github.com/vblagoje/openapi-llm.git

2. Install hatch if you haven't already

pip install hatch

3. Install pre-commit hooks

pre-commit install

4. Install desired LLM provider dependencies (as needed)

pip install openai anthropic cohere

Testing

Run tests using hatch:

# Unit tests
hatch run test:unit

# Integration tests
hatch run test:integration

# Type checking
hatch run test:typing

# Linting
hatch run test:lint

License

MIT License - See LICENSE for details.

Security

For security concerns, please see our Security Policy.

Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

Author

Vladimir Blagojevic ([e-mail])

Reviews and guidance by Madeesh Kannan