feature(#32): fixed selection_item function to get only its link from the list with the format. And adjustify its temperature.

src/common/assembler.py

@@ -48,3 +48,13 @@ class Assembler:
         contact_model = ContactModel()
         contact_model.get_contact_model(data)
         return contact_model
+
+    """mapping result type into json
+    {
+        "program": sms | contacts | browser | select_item_detail_info, 
+        "content": string
+    }
+    """
+
+    def to_result_format(self, program: str, content: str) -> Any:
+        return {"program": program, "content": content}

src/common/program_type.py

@@ -0,0 +1,23 @@
+"""Program Type for all commands to categorize"""
+
+
+class ProgramType:
+    BROWSER = "browser"
+    ALERT = "alert"
+    IMAGE = "image"
+    SMS = "sms"
+    CONTACT = "contact"
+    MESSAGE = "message"
+
+    class BrowserType:
+        OPEN_TAB = "open_tab"
+        OPEN_TAB_SEARCH = "open_tab_search"
+        CLOSE_TAB = "close_tab"
+        PREVIOUS_PAGE = "previous_page"
+        NEXT_PAGE = "next_page"
+        SCROLL_UP = "scroll_up"
+        SCROLL_DOWN = "scroll_down"
+        SCROLL_TOP = "scroll_top"
+        SCROLL_BOTTOM = "scroll_bottom"
+        SELECT_ITEM_DETAIL_INFO = "select_item_detail_info"
+        SELECT_ITEM = "select_item"

src/rising_plugin/risingplugin.py

@@ -126,10 +126,9 @@ def getCompletionOnly(
     query: str,
     model: str = "gpt-4",
 ) -> str:
-    llm = ChatOpenAI(model_name=model, temperature=1.7, openai_api_key=OPENAI_API_KEY)
+    llm = ChatOpenAI(model_name=model, temperature=0.5, openai_api_key=OPENAI_API_KEY)
     chain = load_qa_chain(llm, chain_type="stuff")
-    test_question = """Please return the link of best relatedness of item which the title is "Android Studio in browser" from the below data. [{"title": "Android Studio", "link": "https://android.com"} , {"title": "What's this?", "link": "https://test.com"} , {"title": "How are you?", "link": "https://d.com"}]"""
-    chain_data = chain.run(input_documents=[], question=test_question)
+    chain_data = chain.run(input_documents=[], question=query)
     return chain_data
 
 

src/router/browser_router.py

@@ -1,6 +1,7 @@
 from fastapi import APIRouter, Request, Depends
 
 from src.common.assembler import Assembler
+from src.common.program_type import ProgramType
 from src.common.utils import parseUrlFromStr
 from src.model.requests.request_model import BrowserItem
 from src.service.browser_service import BrowserService
@@ -38,7 +39,12 @@ def construct_blueprint_browser_api() -> APIRouter:
         except Exception as e:
             return assembler.to_response(400, "Failed to get item in a browser", "")
         return assembler.to_response(
-            200, "Getting an item in a browser successfully", parseUrlFromStr(item_link)
+            200,
+            "Getting an item in a browser successfully",
+            assembler.to_result_format(
+                ProgramType.BrowserType.SELECT_ITEM,
+                parseUrlFromStr(item_link),
+            ),
         )
 
     return router

src/service/browser_service.py

@@ -7,5 +7,7 @@ class BrowserService:
     """query to get the link of the item from the list"""
 
     def query_item(self, items: list[BrowserItem.ItemReq], query: str) -> str:
-        prompt_template = f"Please return the link of best relatedness of item which the title is '{query}' from the below data.\n {items}"
+        prompt_template = f"""
+        User is trying to '{query}' and it includes the title of the item.
+        Please return the link of best relatedness of the item with the title from the below data.\n {items}"""
         return getCompletionOnly(query=prompt_template)