chore: github hot project api, allow use streaming

soulteary · May 29, 2023 · 574f3f4 · 574f3f4
1 parent 0066a06
commit 574f3f4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/components/stream-responser/stream_builder.go b/components/stream-responser/stream_builder.go
@@ -5,6 +5,7 @@ import (
 
 	eb "github.com/soulteary/sparrow/components/event-broker"
 	FlagStudio "github.com/soulteary/sparrow/connectors/flag-studio"
+	GitHubTop "github.com/soulteary/sparrow/connectors/github-top"
 	OpenaiAPI "github.com/soulteary/sparrow/connectors/openai-api"
 	"github.com/soulteary/sparrow/internal/datatypes"
 	"github.com/soulteary/sparrow/internal/define"
@@ -52,6 +53,11 @@ func StreamBuilder(parentMessageID string, conversationID string, modelSlug stri
 			sequences = MakeStreamingMessage(input, modelSlug, conversationID, messageID, mode)
 			quickMode = true
 		}
+	case datatypes.MODEL_GITHUB_TOP.Slug:
+		if define.ENABLE_GITHUB_TOP {
+			sequences = MakeStreamingMessage(GitHubTop.HandleUserPrompt(input), modelSlug, conversationID, messageID, mode)
+			quickMode = true
+		}
 		// case datatypes.MODEL_NO_MODELS.Slug:
 		// default:
 	}