openapi: 3.1.0
info:
  title: PDF RAG API
  version: 1.0.0
  description: API to query a Weaviate vector DB using SentenceTransformer embeddings

servers:
  - url: https://vignesh45--pdf-rag-api.hf.space

paths:
  /:
    get:
      summary: Health check
      operationId: root
      responses:
        '200':
          description: OK
          content:
            application/json:
              schema:
                type: object
                properties:
                  message:
                    type: string

    post:
      summary: Query the vector DB with a text prompt
      operationId: queryWeaviate
      requestBody:
        required: true
        content:
          application/json:
            schema:
              type: object
              properties:
                text:
                  type: string
                  minLength: 1
                top_k:
                  type: integer
                  default: 7
                  minimum: 1
                  maximum: 100
              required:
                - text
      responses:
        '200':
          description: Query results
          content:
            application/json:
              schema:
                type: object
                properties:
                  query:
                    type: string
                  results:
                    type: array
                    items:
                      type: object
                      properties:
                        text:
                          type: string
                        source:
                          type: string
        '400':
          description: Bad request
        '500':
          description: Internal server error