Compare commits

...

3 Commits

Author SHA1 Message Date
bf49c3c15d add missing code 2023-08-28 18:21:38 +08:00
7305de467e feat: add passport code 2023-08-28 18:21:13 +08:00
4fa6b2c2bd just add 2023-08-28 18:20:07 +08:00
6 changed files with 59 additions and 2 deletions

21
app/api/passport/route.ts Normal file
View File

@ -0,0 +1,21 @@
import { type NextRequest } from 'next/server'
import { client } from '@/app/api/utils/common'
import { API_KEY, API_URL, APP_ID } from '@/config'
// import { commonClient } from 'dify-client'
import axios from "axios";
export async function GET(request: NextRequest) {
const headers = {
Authorization: `Bearer ${API_KEY}`,
"Content-Type": "application/json",
'X-App-Code': APP_ID
};
const res = await axios({
url: 'https://api.dify.ai/v1/passport',
headers,
responseType: "json",
})
console.log(res)
return new Response(res.data)
}

View File

@ -1,5 +1,5 @@
import { type NextRequest } from 'next/server'
import { ChatClient } from 'dify-client'
import { ChatClient, DifyClient } from 'dify-client'
import { v4 } from 'uuid'
import { API_KEY, API_URL, APP_ID } from '@/config'
@ -19,3 +19,4 @@ export const setSession = (sessionId: string) => {
}
export const client = new ChatClient(API_KEY, API_URL || undefined)
export const commonClient = new ChatClient(API_KEY, API_URL || undefined)

View File

@ -19,6 +19,7 @@ import Loading from '@/app/components/base/loading'
import { replaceVarWithValues, userInputsFormToPromptVariables } from '@/utils/prompt'
import AppUnavailable from '@/app/components/app-unavailable'
import { API_KEY, APP_ID, APP_INFO, isShowPrompt, promptTemplate } from '@/config'
import { checkOrSetAccessToken } from '@/utils/access-token'
const Main: FC = () => {
const { t } = useTranslation()
@ -197,6 +198,8 @@ const Main: FC = () => {
return
}
(async () => {
await checkOrSetAccessToken()
try {
const [conversationData, appParams] = await Promise.all([fetchConversations(), fetchAppParams()])

View File

@ -1,4 +1,4 @@
import { API_PREFIX } from '@/config'
import { API_PREFIX, APP_ID } from '@/config'
import Toast from '@/app/components/base/toast'
const TIME_OUT = 100000
@ -102,7 +102,16 @@ const handleStream = (response: any, onData: IOnData, onCompleted?: IOnCompleted
const baseFetch = (url: string, fetchOptions: any, { needAllResponseContent }: IOtherOptions) => {
const options = Object.assign({}, baseOptions, fetchOptions)
const sharedToken = APP_ID
const accessToken = localStorage.getItem('token') || JSON.stringify({ [sharedToken]: '' })
let accessTokenJson = { [sharedToken]: '' }
try {
accessTokenJson = JSON.parse(accessToken)
}
catch (e) {
}
options.headers.set('Authorization', `Bearer ${accessTokenJson[sharedToken]}`)
const urlPrefix = API_PREFIX
let urlWithPrefix = `${urlPrefix}${url.startsWith('/') ? url : `/${url}`}`

View File

@ -31,3 +31,7 @@ export const fetchAppParams = async () => {
export const updateFeedback = async ({ url, body }: { url: string; body: Feedbacktype }) => {
return post(url, { body })
}
export const fetchAccessToken = async (appId: string) => {
return get('/passport') as Promise<{ access_token: string }>
}

19
utils/access-token.ts Normal file
View File

@ -0,0 +1,19 @@
import { fetchAccessToken } from '@/service'
import { APP_ID } from '@/config'
export const checkOrSetAccessToken = async () => {
const sharedToken = APP_ID
const accessToken = localStorage.getItem('token') || JSON.stringify({ [sharedToken]: '' })
let accessTokenJson = { [sharedToken]: '' }
try {
accessTokenJson = JSON.parse(accessToken)
}
catch (e) {
}
if (!accessTokenJson[sharedToken]) {
const res = await fetchAccessToken(sharedToken)
accessTokenJson[sharedToken] = res.access_token
localStorage.setItem('token', JSON.stringify(accessTokenJson))
}
}