import { Gateway } from "@adaline/gateway";

const gateway = new Gateway({
  cache: myCache,        // optional: pluggable cache backend
  httpClient: myClient,  // optional: custom HTTP client
  logger: myLogger,      // optional: custom logger
});

const response = await gateway.completeChat({
  model: gpt4o,
  config: Config().parse({ temperature: 0.7, maxTokens: 500 }),
  messages: [
    { role: "system", content: [{ modality: "text", value: "You are a helpful assistant." }] },
    { role: "user", content: [{ modality: "text", value: "Explain quantum computing." }] },
  ],
  tools: [],
});

for await (const chunk of gateway.streamChat({
  model: gpt4o,
  config: Config().parse({ temperature: 0.7, maxTokens: 500 }),
  messages: [
    { role: "user", content: [{ modality: "text", value: "Write a poem." }] },
  ],
  tools: [],
})) {
  process.stdout.write(chunk.response);
}

const embeddings = await gateway.getEmbeddings({
  model: textEmbedding3Large,
  config: Config().parse({ encodingFormat: "float", dimensions: 256 }),
  embeddingRequests: {
    modality: "text",
    requests: ["Hello world", "How are you?"],
  },
});

const toolResponses = await gateway.getToolResponses({
  tools: myToolDefinitions,
  toolCalls: response.toolCalls,
});

const cost = Gateway.getChatUsageCost({
  model: gpt4o,
  usage: response.usage,
});