jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: Refactor API key header and remove streaming (#4427)

Browse source 
refactor: Update API key header in get-js-api-code.tsx and remove streaming
This commit is contained in:
anovazzi1 2024-11-06 15:48:07 -03:00 • committed by GitHub
commit 0c049deee3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 51 deletions
Download patch file Download diff file Expand all files Collapse all files

61
src/frontend/src/modals/apiModal/utils/get-js-api-code.tsx
View file

@ -26,8 +26,8 @@ export default function getJsApiCode({
}
async post(endpoint, body, headers = {"Content-Type": "application/json"}) {
if (this.apiKey) {
headers["Authorization"] = \`Bearer \${this.apiKey}\`;
if (this.apiKey) {
headers["x-api-key"] = \`\${this.apiKey}\`;
}
const url = \`\${this.baseURL}\${endpoint}\`;
try {
@ -48,59 +48,22 @@ export default function getJsApiCode({
}
}
async initiateSession(flowId, inputValue, inputType = 'chat', outputType = 'chat', stream = false, tweaks = {}) {
const endpoint = \`/api/v1/run/\${flowId}?stream=\${stream}\`;
async initiateSession(flowId, inputValue, inputType = 'chat', outputType = 'chat', tweaks = {}) {
const endpoint = \`/api/v1/run/\${flowId}\`;
return this.post(endpoint, { ${activeTweaks ? "" : "input_value: inputValue, "}input_type: inputType, output_type: outputType, tweaks: tweaks });
}
async handleStream(streamUrl, onUpdate, onClose, onError) {
try {
const response = await fetch(streamUrl);
const reader = response.body.getReader();
const decoder = new TextDecoder();
while (true) {
const { done, value } = await reader.read();
if (done) {
onClose('Stream closed');
break;
}
const chunk = decoder.decode(value);
const lines = chunk.split(\'\\n\').filter(line => line.trim() !== '');
for (const line of lines) {
if (line.startsWith('data: ')) {
try {
const data = JSON.parse(line.slice(6));
onUpdate(data);
} catch (error) {
console.error('Error parsing JSON:', error);
}
}
}
}
} catch (error) {
console.error('Stream Error:', error);
onError(error);
}
}
async runFlow(flowIdOrName, inputValue, inputType = 'chat', outputType = 'chat', tweaks, stream = false, onUpdate, onClose, onError) {
async runFlow(flowIdOrName, inputValue, inputType = 'chat', outputType = 'chat', tweaks, onError) {
try {
const initResponse = await this.initiateSession(flowIdOrName, inputValue, inputType, outputType, stream, tweaks);
if (stream && initResponse?.outputs?.[0]?.outputs?.[0]?.artifacts?.stream_url) {
const streamUrl = this.baseURL + initResponse.outputs[0].outputs[0].artifacts.stream_url;
console.log(\`Streaming from: \${streamUrl}\`);
this.handleStream(streamUrl, onUpdate, onClose, onError);
}
const initResponse = await this.initiateSession(flowIdOrName, inputValue, inputType, outputType, tweaks);
return initResponse;
} catch (error) {
onError('Error initiating session');
onError('Error initiating session');
}
}
}
async function main(inputValue, inputType = 'chat', outputType = 'chat', stream = false) {
async function main(inputValue, inputType = 'chat', outputType = 'chat') {
const flowIdOrName = '${endpointName || flowId}';
const langflowClient = new LangflowClient('${window.location.protocol}//${window.location.host}',
${isAuth ? "'your-api-key'" : "null"});
@ -113,13 +76,10 @@ export default function getJsApiCode({
inputType,
outputType,
tweaks,
stream,
(data) => console.log("Received:", data.chunk), // onUpdate
(message) => console.log("Stream Closed:", message), // onClose
(error) => console.error("Stream Error:", error) // onError
(error) => console.error("Error:", error) // onError
);
if (!stream && response) {
if (response) {
const flowOutputs = response.outputs[0];
const firstComponentOutputs = flowOutputs.outputs[0];
const output = firstComponentOutputs.outputs.message;
@ -136,7 +96,6 @@ export default function getJsApiCode({
args[0], // inputValue
args[1], // inputType
args[2], // outputType
args[3] === 'true' // streaming
);
`;
}