chore: update Image2Image and Upscale OS storage to use requestID sim…

…ilar to Text2Image and Image2Video (#3092)
livepeer · Jul 4, 2024 · f2c9bb6 · f2c9bb6
1 parent b2933cf
commit f2c9bb6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/ai_mediaserver.go b/server/ai_mediaserver.go
@@ -137,7 +137,7 @@ func (ls *LivepeerServer) ImageToImage() http.Handler {
 
 		params := aiRequestParams{
 			node:        ls.LivepeerNode,
-			os:          drivers.NodeStorage.NewSession(string(core.RandomManifestID())),
+			os:          drivers.NodeStorage.NewSession(requestID),
 			sessManager: ls.AISessionManager,
 		}
 
@@ -295,7 +295,7 @@ func (ls *LivepeerServer) Upscale() http.Handler {
 
 		params := aiRequestParams{
 			node:        ls.LivepeerNode,
-			os:          drivers.NodeStorage.NewSession(string(core.RandomManifestID())),
+			os:          drivers.NodeStorage.NewSession(requestID),
 			sessManager: ls.AISessionManager,
 		}