feat: init

app/api/sdk.js

@@ -0,0 +1,102 @@
+import axios from 'axios'
+
+export class LangGeniusClient {
+  constructor(apiKey, baseUrl = 'https://api.langgenius.ai/v1') {
+    this.apiKey = apiKey
+    this.baseUrl = baseUrl
+  }
+
+  async sendRequest(method, endpoint, data = null, params = null, stream = false) {
+    const headers = {
+      'Authorization': `Bearer ${this.apiKey}`,
+      'Content-Type': 'application/json',
+    }
+
+    const url = `${this.baseUrl}${endpoint}`
+    let response
+    if (!stream) {
+      response = await axios({
+        method,
+        url,
+        data,
+        params,
+        headers,
+        responseType: stream ? 'stream' : 'json',
+      })
+    } else {
+      response = await fetch(url, {
+        headers,
+        method,
+        body: JSON.stringify(data),
+      })
+    }
+
+    return response
+  }
+
+  messageFeedback(messageId, rating, user) {
+    const data = {
+      rating,
+      user,
+    }
+    return this.sendRequest('POST', `/messages/${messageId}/feedbacks`, data)
+  }
+
+  getApplicationParameters(user) {
+    const params = { user }
+    return this.sendRequest('GET', '/parameters', null, params)
+  }
+}
+
+export class CompletionClient extends LangGeniusClient {
+  createCompletionMessage(inputs, query, responseMode, user) {
+    const data = {
+      inputs,
+      query,
+      responseMode,
+      user,
+    }
+    return this.sendRequest('POST', '/completion-messages', data, null, responseMode === 'streaming')
+  }
+}
+
+export class ChatClient extends LangGeniusClient {
+  createChatMessage(inputs, query, user, responseMode = 'blocking', conversationId = null) {
+    const data = {
+      inputs,
+      query,
+      user,
+      responseMode,
+    }
+    if (conversationId)
+      data.conversation_id = conversationId
+
+    return this.sendRequest('POST', '/chat-messages', data, null, responseMode === 'streaming')
+  }
+
+  getConversationMessages(user, conversationId = '', firstId = null, limit = null) {
+    const params = { user }
+
+    if (conversationId)
+      params.conversation_id = conversationId
+
+    if (firstId)
+      params.first_id = firstId
+
+    if (limit)
+      params.limit = limit
+
+    return this.sendRequest('GET', '/messages', null, params)
+  }
+
+  getConversations(user, firstId = null, limit = null, pinned = null) {
+    const params = { user, first_id: firstId, limit, pinned }
+    return this.sendRequest('GET', '/conversations', null, params)
+  }
+
+  renameConversation(conversationId, name, user) {
+    const data = { name, user }
+    return this.sendRequest('PATCH', `/conversations/${conversationId}`, data)
+  }
+}
+