-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
9 changed files
with
244 additions
and
65 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,154 @@ | ||
| package dify | ||
|
|
||
| import ( | ||
| "bufio" | ||
| "bytes" | ||
| "encoding/json" | ||
| "fmt" | ||
| "io" | ||
| "net/http" | ||
| "time" | ||
| ) | ||
|
|
||
| type Dify struct { | ||
| BaseURL string | ||
| SummarizerAppToken string | ||
| } | ||
|
|
||
| // New function return dify service | ||
| func New(baseURL, summarizerAppToken string) DifyAdapter { | ||
| return &Dify{ | ||
| BaseURL: baseURL, | ||
| SummarizerAppToken: summarizerAppToken, | ||
| } | ||
| } | ||
|
|
||
| // BaseEvent represents the common fields in the event | ||
| type BaseEvent struct { | ||
| Event string `json:"event,omitempty"` | ||
| ConversationID string `json:"conversation_id,omitempty"` | ||
| MessageID string `json:"message_id,omitempty"` | ||
| CreatedAt int64 `json:"created_at,omitempty"` | ||
| TaskID string `json:"task_id,omitempty"` | ||
| ID string `json:"id,omitempty"` | ||
| Position int `json:"position,omitempty"` | ||
| } | ||
|
|
||
| // AgentThought represents the specific fields for agent_thought events | ||
| type AgentThought struct { | ||
| BaseEvent | ||
| Thought string `json:"thought,omitempty"` | ||
| Observation string `json:"observation,omitempty"` | ||
| Tool string `json:"tool,omitempty"` | ||
| ToolLabels interface{} `json:"tool_labels,omitempty"` | ||
| ToolInput string `json:"tool_input,omitempty"` | ||
| MessageFiles interface{} `json:"message_files,omitempty"` | ||
| } | ||
|
|
||
| // AgentMessage represents the specific fields for agent_message events | ||
| type AgentMessage struct { | ||
| BaseEvent | ||
| Answer string `json:"answer,omitempty"` | ||
| } | ||
|
|
||
| func (d *Dify) SummarizeArticle(url string) (content string, err error) { | ||
| // Define the URL and request body | ||
| requestBody, err := json.Marshal(map[string]interface{}{ | ||
| "inputs": map[string]interface{}{}, | ||
| "query": url, | ||
| "response_mode": "streaming", | ||
| "conversation_id": "", | ||
| "user": "fortress", | ||
| }) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| // Create the HTTP request | ||
| req, err := http.NewRequest(http.MethodPost, d.BaseURL, bytes.NewBuffer(requestBody)) | ||
| if err != nil { | ||
| return "", nil | ||
| } | ||
|
|
||
| // Set the required headers | ||
| req.Header.Set("Authorization", "Bearer "+d.SummarizerAppToken) | ||
| req.Header.Set("Content-Type", "application/json") | ||
|
|
||
| client := &http.Client{ | ||
| Timeout: 5 * time.Minute, | ||
| } | ||
|
|
||
| resp, err := client.Do(req) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| defer resp.Body.Close() | ||
|
|
||
| // Check the response status | ||
| if resp.StatusCode != http.StatusOK { | ||
| return "", fmt.Errorf("unexpected status code: %d", resp.StatusCode) | ||
| } | ||
|
|
||
| // Handle streaming response | ||
| var thoughts []AgentThought | ||
| reader := bufio.NewReader(resp.Body) | ||
| for { | ||
| line, err := reader.ReadBytes('\n') | ||
| if err != nil { | ||
| if err == io.EOF { | ||
| break | ||
| } | ||
| return "", err | ||
| } | ||
|
|
||
| // Remove the "data: " prefix | ||
| line = bytes.TrimPrefix(line, []byte("data: ")) | ||
| line = bytes.TrimSpace(line) | ||
| if len(line) == 0 { | ||
| continue | ||
| } | ||
|
|
||
| // Parse the JSON event to a map to determine the event type | ||
| var rawEvent map[string]interface{} | ||
| err = json.Unmarshal(line, &rawEvent) | ||
| if err != nil { | ||
| fmt.Printf("Error unmarshal: %v\n", string(line)) | ||
| continue | ||
| } | ||
|
|
||
| eventType, ok := rawEvent["event"].(string) | ||
| if !ok { | ||
| fmt.Println("Error: event type is missing or not a string") | ||
| continue | ||
| } | ||
|
|
||
| // Process specific event types | ||
| switch eventType { | ||
| case "agent_thought": | ||
| var event AgentThought | ||
| err = json.Unmarshal(line, &event) | ||
| if err != nil { | ||
| fmt.Printf("Error parsing agent_thought JSON: %v\n", err) | ||
| continue | ||
| } | ||
| thoughts = append(thoughts, event) | ||
| case "agent_message": | ||
| // just ignore agent_message event | ||
| default: | ||
| } | ||
| } | ||
|
|
||
| // get the last event | ||
| if len(thoughts) == 0 { | ||
| return "", fmt.Errorf("no thought found") | ||
| } | ||
|
|
||
| for i := len(thoughts) - 1; i >= 0; i-- { | ||
| if thoughts[i].Thought != "" { | ||
| content = thoughts[i].Thought | ||
| break | ||
| } | ||
| } | ||
|
|
||
| return content, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,5 @@ | ||
| package dify | ||
|
|
||
| type DifyAdapter interface { | ||
| SummarizeArticle(youtubeURL string) (content string, err error) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.