Feat/sdk vision support (#1531)
Co-authored-by: Joel <iamjoel007@gmail.com>
This commit is contained in:
@@ -19,6 +19,13 @@ class DifyClient {
|
||||
'Content-Type' => 'application/json',
|
||||
],
|
||||
]);
|
||||
$this->file_client = new Client([
|
||||
'base_uri' => $this->base_url,
|
||||
'headers' => [
|
||||
'Authorization' => 'Bearer ' . $this->api_key,
|
||||
'Content-Type' => 'multipart/form-data',
|
||||
],
|
||||
]);
|
||||
}
|
||||
|
||||
protected function send_request($method, $endpoint, $data = null, $params = null, $stream = false) {
|
||||
@@ -44,27 +51,57 @@ class DifyClient {
|
||||
$params = ['user' => $user];
|
||||
return $this->send_request('GET', 'parameters', null, $params);
|
||||
}
|
||||
|
||||
public function file_upload($user, $files) {
|
||||
$data = ['user' => $user];
|
||||
$options = [
|
||||
'multipart' => $this->prepareMultipart($data, $files)
|
||||
];
|
||||
|
||||
return $this->file_client->request('POST', 'files/upload', $options);
|
||||
}
|
||||
|
||||
protected function prepareMultipart($data, $files) {
|
||||
$multipart = [];
|
||||
foreach ($data as $key => $value) {
|
||||
$multipart[] = [
|
||||
'name' => $key,
|
||||
'contents' => $value
|
||||
];
|
||||
}
|
||||
|
||||
foreach ($files as $file) {
|
||||
$multipart[] = [
|
||||
'name' => 'file',
|
||||
'contents' => fopen($file['tmp_name'], 'r'),
|
||||
'filename' => $file['name']
|
||||
];
|
||||
}
|
||||
|
||||
return $multipart;
|
||||
}
|
||||
}
|
||||
|
||||
class CompletionClient extends DifyClient {
|
||||
public function create_completion_message($inputs, $query, $response_mode, $user) {
|
||||
public function create_completion_message($inputs, $response_mode, $user, $files = null) {
|
||||
$data = [
|
||||
'inputs' => $inputs,
|
||||
'query' => $query,
|
||||
'response_mode' => $response_mode,
|
||||
'user' => $user,
|
||||
'files' => $files,
|
||||
];
|
||||
return $this->send_request('POST', 'completion-messages', $data, null, $response_mode === 'streaming');
|
||||
}
|
||||
}
|
||||
|
||||
class ChatClient extends DifyClient {
|
||||
public function create_chat_message($inputs, $query, $user, $response_mode = 'blocking', $conversation_id = null) {
|
||||
public function create_chat_message($inputs, $query, $user, $response_mode = 'blocking', $conversation_id = null, $files = null) {
|
||||
$data = [
|
||||
'inputs' => $inputs,
|
||||
'query' => $query,
|
||||
'user' => $user,
|
||||
'response_mode' => $response_mode,
|
||||
'files' => $files,
|
||||
];
|
||||
if ($conversation_id) {
|
||||
$data['conversation_id'] = $conversation_id;
|
||||
|
Reference in New Issue
Block a user