fix: set generated kube lol

pilksoc · Mar 15, 2024 · a254410 · a254410
1 parent 769b683
commit a254410
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/kube_cache/aiStuff/ai.go b/kube_cache/aiStuff/ai.go
@@ -174,8 +174,12 @@ func (ai *KubeAi) GenerateKubeRecipe(kubeName1, kubeName2 string) (string, error
 			ai.Lock.Unlock()
 			break
 		}
+
 		ai.Lock.Unlock()
-		time.Sleep(apiRestTime - time.Since(ai.LastAccess))
+    sleepTime := apiRestTime - time.Since(ai.LastAccess)
+
+    log.Printf("Rate limited, going to bed for %dms...", sleepTime / time.Millisecond)
+		time.Sleep(sleepTime)
 	}
 
 	res, err := ai.generateKubeRecipe(kubeName1, kubeName2)

diff --git a/kube_cache/server/cache.go b/kube_cache/server/cache.go
@@ -143,6 +143,10 @@ func (s *Server) craft(c *gin.Context, id1, id2 string) (model.KubeRecipe, error
 			return model.KubeRecipe{}, errors.New("Cannot generate kube recipe")
 		}
 
+		recipe, err = s.Database.GetKubeRecipe(id1, id2)
+		if err != nil {
+			log.Printf("Cannot get kube recipe: %s", err)
+		}
 		log.Printf("Generated new kube: %s, generating image in new thread", newKubeId)
 		go func() {
 			image, err := s.Ai.GenerateDalleForKube(newKubeId)
@@ -154,11 +158,6 @@ func (s *Server) craft(c *gin.Context, id1, id2 string) (model.KubeRecipe, error
 			if err != nil {
 				log.Printf("Cannot save kube recipe: %s", err)
 			}
-
-			recipe, err = s.Database.GetKubeRecipe(id1, id2)
-			if err != nil {
-				log.Printf("Cannot get kube recipe: %s", err)
-			}
 		}()
 
 	} else {