Skip to content

Commit

Permalink
Merge pull request #309 from uezo/support-dify
Browse files Browse the repository at this point in the history
Add support for Dify💙
  • Loading branch information
uezo committed Jul 1, 2024
2 parents 55a07fb + 7adf479 commit e51c499
Show file tree
Hide file tree
Showing 7 changed files with 714 additions and 0 deletions.
45 changes: 45 additions & 0 deletions Plugins/DifyServiceWebGL.jslib
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
mergeInto(LibraryManager.library, {
StartDifyMessageStreamJS: function(targetObjectNamePtr, sessionIdPtr, urlPtr, apiKeyPtr, userPtr, difyStreamRequestPtr) {
let targetObjectName = UTF8ToString(targetObjectNamePtr);
let sessionId = UTF8ToString(sessionIdPtr);
let url = UTF8ToString(urlPtr);
let apiKey = UTF8ToString(apiKeyPtr);
let user = UTF8ToString(userPtr);
let difyStreamRequest = UTF8ToString(difyStreamRequestPtr);
let decoder = new TextDecoder("utf-8");

if (document.difyAbortController == null) {
document.difyAbortController = new AbortController();
}

fetch(url, {
headers: {
"Content-Type": "application/json",
"Authorization": `Bearer ${apiKey}`
},
method: "POST",
body: difyStreamRequest,
signal: document.difyAbortController.signal
})
.then((response) => response.body.getReader())
.then((reader) => {
const readChunk = function({done, value}) {
if(done) {
reader.releaseLock();
return;
}
SendMessage(targetObjectName, "SetDifyMessageStreamChunk", sessionId + "::" + decoder.decode(value));
reader.read().then(readChunk);
}
reader.read().then(readChunk);
})
.catch((err) => {
console.error(`Error at fetch: ${err.message}`);
});
},

AbortDifyMessageStreamJS: function() {
console.log("Abort Dify at AbortDifyMessageStreamJS");
document.difyAbortController.abort();
}
});
32 changes: 32 additions & 0 deletions Plugins/DifyServiceWebGL.jslib.meta

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

8 changes: 8 additions & 0 deletions Scripts/LLM/Dify.meta

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading

0 comments on commit e51c499

Please sign in to comment.