inference refactor

Author: l-trotta

output/openapi/elasticsearch-openapi.json

@@ -18,18 +18,17 @@
  */
 
 import { RequestChatCompletionBase } from '@inference/_types/CommonTypes'
-import { UserDefinedValue } from '@spec_utils/UserDefinedValue'
+import { RequestBase } from '@_types/Base'
 import { Id } from '@_types/common'
 import { Duration } from '@_types/Time'
-
 /**
  * Perform chat completion inference
  * @rest_spec_name inference.chat_completion_unified
  * @availability stack since=8.18.0 stability=stable visibility=public
  * @availability serverless stability=stable visibility=public
  * @doc_id inference-api-chat-completion
  */
-export interface Request extends RequestChatCompletionBase {
+export interface Request extends RequestBase {
   urls: [
     {
       path: '/_inference/chat_completion/{inference_id}/_stream'
@@ -49,144 +48,6 @@ export interface Request extends RequestChatCompletionBase {
      */
     timeout?: Duration
   }
-}
-
-/**
- * @codegen_names string, object
- */
-export type CompletionToolType = string | CompletionToolChoice
-
-/**
- * An object style representation of a single portion of a conversation.
- */
-export interface ContentObject {
-  /**
-   * The text content.
-   */
-  text: string
-  /**
-   * The type of content.
-   */
-  type: string
-}
-
-/**
- * The function that the model called.
- */
-export interface ToolCallFunction {
-  /**
-   * The arguments to call the function with in JSON format.
-   */
-  arguments: string
-  /**
-   * The name of the function to call.
-   */
-  name: string
-}
-
-/**
- * A tool call generated by the model.
- */
-export interface ToolCall {
-  /**
-   * The identifier of the tool call.
-   */
-  id: Id
-  /**
-   * The function that the model called.
-   */
-  function: ToolCallFunction
-  /**
-   * The type of the tool call.
-   */
-  type: string
-}
-
-/**
- * @codegen_names string, object
- */
-export type MessageContent = string | Array<ContentObject>
-
-/**
- * An object representing part of the conversation.
- */
-export interface Message {
-  /**
-   * The content of the message.
-   */
-  content?: MessageContent
-  /**
-   * The role of the message author.
-   */
-  role: string
-  /**
-   * The tool call that this message is responding to.
-   */
-  tool_call_id?: Id
-  /**
-   * The tool calls generated by the model.
-   */
-  tool_calls?: Array<ToolCall>
-}
-
-/**
- * The tool choice function.
- *
- */
-export interface CompletionToolChoiceFunction {
-  /**
-   * The name of the function to call.
-   */
-  name: string
-}
-
-/**
- * Controls which tool is called by the model.
- */
-export interface CompletionToolChoice {
-  /**
-   * The type of the tool.
-   */
-  type: string
-  /**
-   * The tool choice function.
-   */
-  function: CompletionToolChoiceFunction
-}
-
-/**
- * The completion tool function definition.
- */
-export interface CompletionToolFunction {
-  /**
-   * A description of what the function does.
-   * This is used by the model to choose when and how to call the function.
-   */
-  description?: string
-  /**
-   * The name of the function.
-   */
-  name: string
-  /**
-   * The parameters the functional accepts. This should be formatted as a JSON object.
-   */
-  parameters?: UserDefinedValue
-  /**
-   * Whether to enable schema adherence when generating the function call.
-   */
-  strict?: boolean
-}
-
-/**
- * A list of tools that the model can call.
- */
-export interface CompletionTool {
-  /**
-   * The type of tool.
-   */
-  type: string
-  /**
-   * The function definition.
-   */
-  function: CompletionToolFunction
+  /** @codegen_name chat_completion_request */
+  body: RequestChatCompletionBase
 }

output/openapi/elasticsearch-serverless-openapi.json

@@ -20,5 +20,6 @@
 import { StreamResult } from '@_types/Binary'
 
 export class Response {
+  /** @codegen_name data */
   body: StreamResult
 }

output/schema/schema-serverless.json

@@ -20,5 +20,6 @@
 import { CompletionInferenceResult } from '@inference/_types/Results'
 
 export class Response {
+  /** @codegen_name completion_result */
   body: CompletionInferenceResult
 }

output/schema/schema.json

@@ -20,5 +20,6 @@
 import { DeleteInferenceEndpointResult } from '@inference/_types/Results'
 
 export class Response {
+  /** @codegen_name delete_result */
   body: DeleteInferenceEndpointResult
 }