scnario commited on
Commit
3c547a5
·
verified ·
1 Parent(s): b74e526

Update server.js

Browse files
Files changed (1) hide show
  1. server.js +38 -0
server.js CHANGED
@@ -2,6 +2,7 @@ const express = require('express');
2
  const axios = require('axios');
3
  const fs = require('fs');
4
  const path = require('path');
 
5
 
6
  const app = express();
7
  app.set('json spaces', 2);
@@ -62,6 +63,43 @@ async function deepseek(sessionId, prompt) {
62
  return response;
63
  }
64
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
65
  app.get('/google/gemini-1.5-pro', async (req, res) => {
66
  try {
67
  const { question, sessionId } = req.query;
 
2
  const axios = require('axios');
3
  const fs = require('fs');
4
  const path = require('path');
5
+ const swaggerUi = require('swagger-ui-express');
6
 
7
  const app = express();
8
  app.set('json spaces', 2);
 
63
  return response;
64
  }
65
 
66
+ const swaggerDocument = {
67
+ openapi: '3.0.0',
68
+ info: { title: 'API Documentation', version: '1.0.0' },
69
+ paths: {
70
+ '/google/gemini-1.5-pro': {
71
+ get: {
72
+ summary: 'Chat with @google/gemini-1.5-pro',
73
+ parameters: [
74
+ { name: 'question', in: 'query', required: true, schema: { type: 'string' } },
75
+ { name: 'sessionId', in: 'query', required: true, schema: { type: 'string' } }
76
+ ],
77
+ responses: {
78
+ 200: { description: 'Successful response', content: { 'application/json': { schema: { type: 'object', properties: { reply: { type: 'string' } } } } } },
79
+ 400: { description: 'Missing parameters' },
80
+ 500: { description: 'Server error' }
81
+ }
82
+ }
83
+ },
84
+ '/deepseek-ai/deepseek-r1-distill-qwen-32b': {
85
+ get: {
86
+ summary: 'Chat with @deepseek-ai/deepseek-r1-distill-qwen-32b',
87
+ parameters: [
88
+ { name: 'question', in: 'query', required: true, schema: { type: 'string' } },
89
+ { name: 'sessionId', in: 'query', required: true, schema: { type: 'string' } }
90
+ ],
91
+ responses: {
92
+ 200: { description: 'Successful response', content: { 'application/json': { schema: { type: 'object', properties: { reply: { type: 'string' }, think: { type: 'string' } } } } } },
93
+ 400: { description: 'Missing parameters' },
94
+ 500: { description: 'Server error' }
95
+ }
96
+ }
97
+ }
98
+ }
99
+ };
100
+
101
+ app.use('/', swaggerUi.serve, swaggerUi.setup(swaggerDocument));
102
+
103
  app.get('/google/gemini-1.5-pro', async (req, res) => {
104
  try {
105
  const { question, sessionId } = req.query;