request model

AUTOMATIC1111 · Jun 7, 2024 · 71110e0 · 71110e0
1 parent 1afd007
commit 71110e0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/modules/api/api.py b/modules/api/api.py
@@ -422,7 +422,7 @@ def extract_nouns(self, sentence):
 
         return nouns
 
-    async def nature2prompt(self, nature2promptreq: models.Nature2PromptResponse):
+    async def nature2prompt(self, nature2promptreq: models.Nature2PromptRequest):
         client = AsyncOpenAI()
 
         n2p_logger = logging.getLogger(__name__)
@@ -437,7 +437,7 @@ async def nature2prompt(self, nature2promptreq: models.Nature2PromptResponse):
                 # {"role": "assistant",
                 #  "content":" ",},
                 {"role": "user",
-                 "content":input_to_bot.filtered_nouns,}
+                 "content":input_to_bot.text,}
                 # {"role": "assistant",
                 #  "content":" ",},
             ],

diff --git a/modules/api/models.py b/modules/api/models.py
@@ -131,6 +131,9 @@ def generate_model(self):
     ]
 ).generate_model()
 
+class Nature2PromptRequest(BaseModel):
+    text: str
+
 class Nature2PromptResponse(BaseModel):
     filtered_nouns: str