fix bug in API

Author: vaayne

README.md

@@ -24,8 +24,38 @@ We support these advanced APIs:
 
 
 1. `writing_with_topic`
+    ```
+    Writing for special topic
+    Args:
+        topic (TopicEnum): the special topic
+        prompt (str): prompt for writing
+    Example:
+        ao = NotionAI(token)
+        ai.writing_with_topic(TopicEnum.blog_post, "Please help to introduce Notion")
+    ```
 2. `writing_with_prompt`
-
+    ```
+    Writing with special prompt, like summarize, explain_this, improve_writing
+
+    Args:
+        prompt_type (PromptTypeEnum): special prompt
+        context (str): the context for your writing
+
+    Example:
+        ai = NotionAI(token)
+        ai.writing_with_prompt(PromptTypeEnum.summarize, "I am a student")
+    ```
+3. `translate`
+    ```
+    Translate the text
+    Args:
+        language (TranslateLanguageEnum): target language
+        context (str): context to translate
+
+    Example:
+        ai = NotionAI(token)
+        ai.translate(TranslateLanguageEnum.Chinese, "I am a student")
+    ```
 
 
 ## Usage

examples/basic/main.py

@@ -1,4 +1,4 @@
-from notionai.notionai import NotionAI, NotionAIStream
+from notionai import NotionAI, NotionAIStream
 import sys
 import os
 

notionai/notionai.py

@@ -101,10 +101,12 @@ def writing_with_prompt(
         Returns:
             str: Response from NotionAI
         """
+        if prompt_type == PromptTypeEnum.continue_writing:
+            return self.continue_write(context, page_title)
+
         if prompt_type in {
             PromptTypeEnum.help_me_write,
             PromptTypeEnum.help_me_edit,
-            PromptTypeEnum.continue_writing,
             PromptTypeEnum.translate,
         }:
             raise ValueError("Please use the specific method for this prompt type")
@@ -235,40 +237,40 @@ def simplify_language(self, context: str, page_title: str = "") -> str:
         )
 
     def blog_post(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.blog_ost, prompt)
+        return self.writing_with_topic(TopicEnum.blog_post, prompt)
 
     def brainsteam(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.brainsteam, prompt)
+        return self.writing_with_topic(TopicEnum.brainsteam, prompt)
 
     def outline(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.outline, prompt)
+        return self.writing_with_topic(TopicEnum.outline, prompt)
 
     def social_media_post(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.social_media_post, prompt)
+        return self.writing_with_topic(TopicEnum.social_media_post, prompt)
 
     def creative_story(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.creative_story, prompt)
+        return self.writing_with_topic(TopicEnum.creative_story, prompt)
 
     def poem(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.poem, prompt)
+        return self.writing_with_topic(TopicEnum.poem, prompt)
 
     def essay(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.essay, prompt)
+        return self.writing_with_topic(TopicEnum.essay, prompt)
 
     def meeting_agenda(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.meeting_agenda, prompt)
+        return self.writing_with_topic(TopicEnum.meeting_agenda, prompt)
 
     def press_release(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.press_release, prompt)
+        return self.writing_with_topic(TopicEnum.press_release, prompt)
 
     def job_description(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.job_description, prompt)
+        return self.writing_with_topic(TopicEnum.job_description, prompt)
 
     def sales_email(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.sales_email, prompt)
+        return self.writing_with_topic(TopicEnum.sales_email, prompt)
 
     def recruiting_email(self, prompt: str) -> str:
-        return self.write_with_topic(TopicEnum.recruiting_email, prompt)
+        return self.writing_with_topic(TopicEnum.recruiting_email, prompt)
 
 
 class NotionAIStream(NotionAI):

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "notionai-py"
 description = "Unoffical Notion AI API"
-version="0.0.4"
+version="0.0.5"
 authors = [
   { name="Vaayne", email="[email protected]" },
 ]