update

.gitignore

@@ -1,3 +1,4 @@
+.vscode
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

.vscode/launch.json

@@ -1,21 +1,19 @@
 {
-    // Use IntelliSense to learn about possible attributes.
-    // Hover to view descriptions of existing attributes.
-    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    // 使用 IntelliSense 了解相关属性。 
+    // 悬停以查看现有属性的描述。
+    // 欲了解更多信息，请访问: https://go.microsoft.com/fwlink/?linkid=830387
     "version": "0.2.0",
     "configurations": [
-        
         {
-            "name": "Python: FastAPI",
-            "type": "python",
+            "name": "Python 调试程序: FastAPI",
+            "type": "debugpy",
             "request": "launch",
             "module": "uvicorn",
             "args": [
-                "main:app",
+                "src.main:app",
                 "--reload"
             ],
-            "jinja": true,
-            "justMyCode": true
-        }
+            "jinja": true
+        },
     ]
 }

Dockerfile

@@ -8,4 +8,4 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . .
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "4"]
+CMD ["uvicorn", "src.main:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "4"]

README.md

@@ -1,5 +1,5 @@
 ---
-title: Pgai
+title: FastAPI
 emoji: 📊
 colorFrom: red
 colorTo: pink

games/angrybirds/add_config.py

@@ -1,63 +0,0 @@
-import copy
-
-from games.angrybirds.config_const import x_bird
-from games.angrybirds.config_const import x_pig
-from games.angrybirds.config_const import x_house
-from games.angrybirds.config_const import x_obstacle
-from games.angrybirds.config_const import x_pig_size
-from games.angrybirds.config_const import x_bird_size
-
-from games.angrybirds.config_const import x_change_position
-from games.angrybirds.config_const import x_change_skin
-from games.angrybirds.config_const import x_add
-from games.angrybirds.config_const import x_delete
-from games.angrybirds.config_const import x_list
-from games.angrybirds.config_const import x_increase
-from games.angrybirds.config_const import x_decrease
-
-from games.angrybirds.config_const import x_quantity
-from games.angrybirds.config_const import x_color
-from games.angrybirds.config_const import x_type
-from games.angrybirds.config_const import x_size
-from games.angrybirds.config_const import x_args
-
-# from games.angrybirds.config import config
-
-def change_attribute(d, attr, value):
-    d['result'][2][attr] = value
-
-def add_config(game_train):
-    ###########################################
-    # additional 'add houses'
-    ###########################################
-    example = 'build houses'
-    assert example in game_train
-    ###########################################
-
-    x = 'protect the pigs by build houses for them'
-    game_train[x] = game_train[example]
-    print(len(game_train))
-
-    x = "build a house"
-    game_train[x] = copy.deepcopy(game_train[example])
-    assert game_train[x]['result'][2][x_quantity] == 'some'
-    change_attribute(game_train[x], x_quantity, 1)
-    print(len(game_train))
-
-    x = "build houses, so I can protect my pigs"
-    game_train[x] = game_train[example]
-    print(len(game_train))
-
-    ###########################################
-    # additional 'delete birds'
-    ###########################################
-    for bird in [
-        'big', 'magnet', 'superheavy',
-        'catapult', 'flash', 'antigravity', 
-    ]:
-        example = f'delete {bird} birds'
-        assert example in game_train
-        ###########################################
-        x = f"I don't like {bird} birds. Delete them"
-        game_train[x] = game_train[example]
-        print(len(game_train))

games/angrybirds/angrybirds.py

@@ -1,172 +0,0 @@
-import json
-from pathlib import Path
-from collections import Counter
-
-import openai
-from games.game import game
-# from games.game_openai import api_key
-from games.angrybirds.config import config
-
-debug = 2
-
-class angrybird (game):
-    ai_version   = '3.8.0.2023-09-18-a' # # use inheritance
-    data_version = ai_version
-
-    def __init__(self):
-        # openai.api_key = api_key
-        self.ROOT = Path(__file__).parent.parent.parent
-        self.PATH = self.__class__.__name__
-        self.embeddings = {}
-        self.query_count = 0
-
-        def init_train():
-            PATH = Path(__file__).parent
-            with open(PATH/"game_train.json", 'r') as f:
-                unnormalized = json.load(f)
-
-            self.train = {}
-            for cmd in unnormalized.keys():
-                key = cmd.strip().lower()
-                self.train[key] = unnormalized[cmd]
-            print(f"{len(self.train)=}")
-
-        # self.train = self.init_train(config)
-        init_train()
-
-
-    def special_case(self, command: str):
-        if command.strip().lower() in ['bird', 'birds']:
-            print('special case')
-            return 'list birds'
-        elif command.strip().lower() in ['pig', 'pigs', 'enemy', 'enemies']:
-            print('special case')
-            return 'list pigs'
-        else:
-            return command
-
-
-    def _get_full_prompt(self, prompt):
-        assistant = {
-            "role": "system", 
-            "content": "You are a helpful assistant.",
-        }
-        full_prompt = [
-            assistant,
-            {'role': 'user', 'content': prompt}
-        ]
-        return full_prompt
-
-
-    def _query_chatgpt(self, full_prompt:str, n:int=15):
-        completion = openai.ChatCompletion.create(
-            # model='gpt-4', 
-            model='gpt-3.5-turbo', 
-            messages=full_prompt, # self._get_full_prompt(prompt),
-            n=n,
-        )
-        choices = [choice.message.content for choice in completion.choices]
-        return choices
-
-
-    def query_chatgpt_nlp_response(
-            self,
-            command: str,   
-            n: int,         
-    ):
-        
-            prompt1 = f"""
-{config.game_desc}
-"""
-            full_prompt = self._get_full_prompt()
-            # print(prompt)
-            choices = self._query_chatgpt(self._get_full_prompt(prompt1), n=1)
-
-
-            prompt2 = f"""
-As a game developer, when I say, 
-
-{command}
-
-give me a concise response.
-"""
-            # print(prompt)
-            choices = self._query_chatgpt(self._get_full_prompt(prompt1), n=1)
-
-            return choices
-
-
-    def query_chatgpt_yes_no(
-            self, 
-            command: str, 
-            follow_up_question: str,
-    ):
-
-        prompt = f"""
-When the user says:
-
-'{command}'
-
-{follow_up_question}
-
-Please answer yes or no.
-Please do not explain.
-"""
-
-        tmp = """
-Please answer yes or no.
-Please do not explain.
-"""
-
-        if debug >= 2:
-            print('-'*10)
-            print(prompt)
-            print('-'*10)
-        choices = self._query_chatgpt(self._get_full_prompt(prompt))
-        # res = completion.choices[0].message.content
-
-        # yes_no
-        for i in range(len(choices)):
-            choices[i] = choices[i].lower()[0]
-            
-        res, freq = Counter(choices).most_common()[0]
-        if debug >= 2:
-            print(f"{freq/len(choices):.4f} {freq}/{len(choices)}\t{res}")
-            print(Counter(choices).most_common())
-        return res
-
-
-    def query_chatgpt_number(
-            self, 
-            command: str, 
-            follow_up_question: str,
-    ):
-
-        prompt = f"""
-When the user says:
-
-'{command}'
-
-{follow_up_question}
-
-Just provide the number in numerical form.
-Please do not explain.
-"""
-
-        tmp = """
-Please answer yes or no.
-Please do not explain.
-"""
-
-        if debug >= 2:
-            print('-'*10)
-            print(prompt)
-            print('-'*10)
-        choices = self._query_chatgpt(self._get_full_prompt(prompt))
-        # res = completion.choices[0].message.content
-
-        res, freq = Counter(choices).most_common()[0]
-        if debug >= 2:
-            print(f"{freq/len(choices):.4f} {freq}/{len(choices)}\t{res}")
-            print(Counter(choices).most_common())
-        return res

games/angrybirds/config.py

@@ -1,646 +0,0 @@
-import json
-
-from games.angrybirds.add_config import add_config
-from games.angrybirds.config_const import *
-
-class config:
-    game_train  = {}
-    agent = 'game developer'
-    game_desc = """
-    Imaging we are developing the game of the Angry Birds, 
-    and the game is described as the following: 
-
-    In Angry Birds, players take control of a group of birds who are seeking 
-    revenge on a group of green pigs that have stolen their eggs. 
-    The gameplay involves using a slingshot to launch the birds at various 
-    structures where the pigs are hiding. The objective is to 
-    eliminate all the pigs on each level and score as many points as possible.
-
-    A game developer is building the game, 
-    and I need you to help me figure out what is going on.
-    """
-
-    def init_game_train(self):
-        ###########################################
-        # add/remove <5> (<antigravity>) birds/pigs
-        ###########################################
-        for obj, obj_single, obj_plural in [
-            (x_bird, 'bird', 'birds'),
-            (x_pig, 'pig', 'pigs'),
-            (x_pig, 'enemy', 'enemies'),
-        ]:
-            for verb1, verb2 in [
-                ('add', x_add),
-                ('delete', x_delete),
-                ('reduce', x_delete),
-            ]:
-                adjs = ['elderly', 'sport',] if obj == x_pig else [
-                    'big', 'magnet', 'superheavy',
-                    'catapult', 'flash', 'antigravity', 
-                ]
-                for certain in adjs + ['']: # empty means no adj
-                    for x1, x2, x3 in [
-                        ('one', obj_single, 1),
-                        ('a' if (certain == '') or (certain[0] not in 'aeiou') else 'an', obj_single, 1),
-                        ('two', obj_plural, 2),
-                        (2, obj_plural, 2),
-                        ('three', obj_plural, 3),
-                        (3, obj_plural, 3),
-                        ('four', obj_plural, 4),
-                        (4, obj_plural, 4),
-                        ('five', obj_plural, 5),
-                        (5, obj_plural, 5),
-                        ('six', obj_plural, 6),
-                        (6, obj_plural, 6),
-                        ('seven', obj_plural, 7),
-                        (7, obj_plural, 7),
-                        ('eight', obj_plural, 8),
-                        (8, obj_plural, 8),
-                        ('nine', obj_plural, 9),
-                        (9, obj_plural, 9),
-                        ('ten', obj_plural, 10),
-                        (10, obj_plural, 10),
-                        ('more', obj_plural, 'some'),
-                        ('some', obj_plural, 'some'),
-                        ('', obj_plural, 'all' if verb2 in [x_delete] else 'some'),
-                        ('all', obj_plural, 'all'),
-                    ]:
-                        tmp = f"{verb1} {x1} {certain} {x2}"
-                        key = ' '.join(tmp.split())
-                        xxx = {
-                            key: {
-                                'train': '',
-                                'result': [
-                                    obj,
-                                    verb2,
-                                    {x_quantity: x3, x_type: certain},
-                                ],
-                                'desc': [
-                                    '[unified] add/delete one or more (special) birds/pigs',
-                                ]
-                            },
-                        }
-                        if certain == '':
-                            del xxx[key]['result'][2][x_type]
-                        # if x3 == 1:
-                        #     xxx[f"add {x1} {x2}"]['result'][2] = {}
-                        self.game_train |= xxx
-
-        ###########################################
-        # add pigs
-        ###########################################
-        for objs in ['pigs', 'enemies']:
-            xxx = {
-                f"make {objs} everywehre": {
-                    'train': '',
-                    'result': [
-                        x_pig,
-                        x_add,
-                        {x_quantity: 'everywhere'},
-                    ],
-                    'desc': [
-                        '[unified] add/delete one or more (special) birds/pigs',
-                    ]
-                },
-                f"Our {objs} are losing. Add more {objs}": {
-                    'train': '',
-                    'result': [
-                        x_pig,
-                        x_add,
-                        {x_quantity: 'some'},
-                    ],
-                    'desc': [
-                        '[unified] add/delete one or more (special) birds/pigs',
-                    ]
-                },
-            }
-            self.game_train |= xxx
-
-        ###########################################
-        # add houses
-        ###########################################
-        xxx = {
-            "build houses": {
-                'train': '',
-                'result': [
-                    x_house,
-                    x_add,
-                    {x_quantity: 'some'},
-                ],
-                'desc': [
-                    'add one or more houses',
-                ]
-            },
-        }
-        self.game_train |= xxx
-
-        ###########################################
-        # add solid houses
-        ###########################################
-        xxx = {
-            "The obstacles are too weak. Let's build some solid houses": {
-                'train': '',
-                'result': [
-                    x_house,
-                    x_add,
-                    {x_type: 'solid', x_quantity: 'some'},
-                ],
-                'desc': [
-                    'add one or more houses',
-                ]
-            },
-        }
-        self.game_train |= xxx
-
-        ###########################################
-        # change HP for pigs
-        ###########################################
-        for obj_single, objs in [
-            ('pig', 'pigs'), 
-            ('enemy', 'enemies'),
-        ]:
-            xxx = {
-                f"The {objs} are too weak. Increase the health points for them": {
-                    'train': '',
-                    'result': [
-                        'pig',
-                        'increase health points',
-                        {},
-                    ],
-                    'desc': [
-                        'change HP',
-                    ]
-                },
-                f"The {objs} are too weak. Increase the HP for them": {
-                    'train': '',
-                    'result': [
-                        'pig',
-                        'increase health points',
-                        {},
-                    ],
-                    'desc': [
-                        'change HP',
-                    ]
-                },
-                f"The {objs} are too weak. Increase the hit points for them": {
-                    'train': '',
-                    'result': [
-                        'pig',
-                        'increase health points',
-                        {},
-                    ],
-                    'desc': [
-                        'change HP',
-                    ]
-                },
-                f"The {objs} are too weak. Increase the blood level for them": {
-                    'train': '',
-                    'result': [
-                        'pig',
-                        'increase health points',
-                        {},
-                    ],
-                    'desc': [
-                        'change HP',
-                    ]
-                },
-                f"The {objs} are too weak. Increase the blood levels for them": {
-                    'train': '',
-                    'result': [
-                        'pig',
-                        'increase health points',
-                        {},
-                    ],
-                    'desc': [
-                        'change HP',
-                    ]
-                },
-                f"make the {obj_single} weaker": {
-                    'train': '',
-                    'result': [
-                        x_pig,
-                        'decrease health points',
-                        {x_quantity: 1},
-                    ],
-                    'desc': [
-                        'change HP',
-                    ]
-                },
-            }
-            self.game_train |= xxx
-
-        ###########################################
-        # Change background
-        ###########################################
-        xxx = {
-            "Change background music": {
-                'train': '',
-                'result': [
-                    'change background',
-                    {x_args: 'music'},
-                ],
-                'desc': [
-                    'change background',
-                ]
-            },
-            "Change background sound": {
-                'train': '',
-                'result': [
-                    'change background',
-                    {x_args: 'music'},
-                ],
-                'desc': [
-                    'change background',
-                ]
-            },
-            "Change background volumn": {
-                'train': '',
-                'result': [
-                    'change background',
-                    {x_args: 'volumn'},
-                ],
-                'desc': [
-                    'change background',
-                ]
-            },
-            "Change volumn": {
-                'train': '',
-                'result': [
-                    'change background',
-                    {x_args: 'volumn'},
-                ],
-                'desc': [
-                    'change background',
-                ]
-            },
-            "Change background": {
-                'train': '',
-                'result': [
-                    'change background',
-                    {x_args: 'picture'},
-                ],
-                'desc': [
-                    'change background',
-                ]
-            },
-            "Change background picture": {
-                'train': '',
-                'result': [
-                    'change background',
-                    {x_args: 'picture'},
-                ],
-                'desc': [
-                    'change background',
-                ]
-            },
-            "Change background photo": {
-                'exclude': '',
-                'result': [
-                    'change background',
-                    {x_args: 'picture'},
-                ],
-                'desc': [
-                    'change background',
-                ]
-            },
-            "Change background color": {
-                'train': '',
-                'result': [
-                    'change background',
-                    {x_args: 'picture'},
-                ],
-                'desc': [
-                    'change background',
-                ]
-            },
-            "Change background text": {
-                'train': '',
-                'result': [
-                    'change background',
-                    {x_args: 'picture'},
-                ],
-                'desc': [
-                    'change background',
-                ]
-            },
-        }
-        self.game_train |= xxx
-
-        ###########################################
-        # change position (birds, pigs, etc)
-        ###########################################
-        for obj, obj_plural in [
-            (x_bird, 'birds'),
-            (x_pig, 'pigs'),
-            (x_pig, 'enemies'),
-            (x_house, 'houses'),
-            (x_obstacle, 'obstacles'),
-        ]:
-            xxx = {
-                f'change the position of {obj_plural}': {
-                    'train': '',
-                    'prompt': [
-                    ],
-                    'result': [
-                        obj,
-                        x_change_position,
-                        {},
-                    ],
-                    'desc': [
-                        f'change positions',
-                    ]
-                },
-                f'move {obj_plural}': {
-                    'train': '',
-                    'prompt': [
-                    ],
-                    'result': [
-                        obj,
-                        x_change_position,
-                        {},
-                    ],
-                    'desc': [
-                        f'change positions',
-                    ]
-                },
-            }
-            self.game_train |= xxx
-
-        ###########################################
-        # change size (birds/pigs)
-        ###########################################
-        xxx = {
-            "Make the bird bigger": {
-                'train': '',
-                'result': [
-                    x_bird_size,
-                    x_increase,
-                    {},
-                ],
-                'desc': [
-                    'change size',
-                ]
-            },
-            "Make the bird bigger, so it can run over the enemies": {
-                'train': '',
-                'focus': '',
-                'result': [
-                    x_bird_size,
-                    x_increase,
-                    {},
-                ],
-                'desc': [
-                    'change size',
-                ]
-            },
-            "Make the pig bigger": {
-                'train': '',
-                'result': [
-                    x_pig_size,
-                    x_increase,
-                    {},
-                ],
-                'desc': [
-                    'change size',
-                ]
-            },
-            "Make the enemy bigger": {
-                'train': '',
-                'result': [
-                    x_pig_size,
-                    x_increase,
-                    {},
-                ],
-                'desc': [
-                    'change size',
-                ]
-            },
-            "Make the bird smaller": {
-                'train': '',
-                'result': [
-                    x_bird_size,
-                    x_decrease,
-                    {},
-                ],
-                'desc': [
-                    'change size',
-                ]
-            },
-            "Make the pig smaller": {
-                'train': '',
-                'result': [
-                    x_pig_size,
-                    x_decrease,
-                    {},
-                ],
-                'desc': [
-                    'change size',
-                ]
-            },
-            "Make the enemy smaller": {
-                'train': '',
-                'result': [
-                    x_pig_size,
-                    x_decrease,
-                    {},
-                ],
-                'desc': [
-                    'change size',
-                ]
-            },
-        }
-        self.game_train |= xxx
-
-        ###########################################
-        # Change skin for birds
-        ###########################################
-        xxx = {
-            "Change bird's picture": {
-                'train': '',
-                'result': [
-                    x_change_skin,
-                    {x_args: 'birds'},
-                ],
-                'desc': [
-                    'change skin',
-                    'change skin for birds',
-                ]
-            },
-            "Change bird's skin": {
-                'train': '',
-                'result': [
-                    x_change_skin,
-                    {x_args: 'birds'},
-                ],
-                'desc': [
-                    'change skin',
-                    'change skin for birds',
-                ]
-            },
-            "bird picture": {
-                'train': '',
-                'result': [
-                    x_change_skin,
-                    {x_args: 'birds'},
-                ],
-                'desc': [
-                    'change skin',
-                    'change skin for bird',
-                ]
-            },
-            "pig picture": {
-                'train': '',
-                'result': [
-                    x_change_skin,
-                    {x_args: 'pigs'},
-                ],
-                'desc': [
-                    'change skin',
-                    'change skin for pigs',
-                ]
-            },
-            "enemy picture": {
-                'train': '',
-                'result': [
-                    x_change_skin,
-                    {x_args: 'pigs'},
-                ],
-                'desc': [
-                    'change skin',
-                    'change skin for pigs',
-                ]
-            },
-            "Change pig's picture": {
-                'train': '',
-                'result': [
-                    x_change_skin,
-                    {x_args: 'pigs'},
-                ],
-                'desc': [
-                    'change skin',
-                    'change skin for pigs',
-                ]
-            },
-            "Change enemy's picture": {
-                'train': '',
-                'result': [
-                    x_change_skin,
-                    {x_args: 'pigs'},
-                ],
-                'desc': [
-                    'change skin',
-                    'change skin for pigs',
-                ]
-            },
-            "Change pig's skin": {
-                'train': '',
-                'result': [
-                    x_change_skin,
-                    {x_args: 'pigs'},
-                ],
-                'desc': [
-                    'change skin',
-                    'change skin for pigs',
-                ]
-            },
-            "Change enemy's skin": {
-                'train': '',
-                'result': [
-                    x_change_skin,
-                    {x_args: 'pigs'},
-                ],
-                'desc': [
-                    'change skin',
-                    'change skin for pigs',
-                ]
-            },
-        }
-        self.game_train |= xxx
-
-        ###########################################
-        # list birds/pigs
-        ###########################################
-        for obj, obj_single, obj_plural in [
-            (x_bird, 'bird', 'birds'),
-            (x_pig, 'pig', 'pigs'),
-            (x_pig, 'enemy', 'enemies'),
-        ]:
-            xxx = {
-                f"List {obj_plural}": {
-                    'train': '',
-                    'result': [
-                        obj,
-                        x_list,
-                        {},
-                    ],
-                    'desc': [
-                        f'list birds or pigs',
-                    ]
-                },
-                f"Show {obj_single} list": {
-                    'train': '',
-                    'result': [
-                        obj,
-                        x_list,
-                        {},
-                    ],
-                    'desc': [
-                        f'list birds or pigs',
-                    ]
-                },
-                f"{obj_single} list": {
-                    'train': '',
-                    'result': [
-                        obj,
-                        x_list,
-                        {},
-                    ],
-                    'desc': [
-                        f'list birds or pigs',
-                    ]
-                },
-                f"What kind of {obj_plural} do you have?": {
-                    'train': '',
-                    'result': [
-                        obj,
-                        x_list,
-                        {},
-                    ],
-                    'desc': [
-                        f'list birds or pigs',
-                    ]
-                },
-                f"Are there new kinds of {obj_plural}?": {
-                    'train': '',
-                    'result': [
-                        obj,
-                        x_list,
-                        {x_type: 'new'},
-                    ],
-                    'desc': [
-                        f'list birds or pigs',
-                    ]
-                },
-                # slight strange semantics
-                f"Change {obj_plural}": {
-                    'train': '',
-                    'result': [
-                        obj,
-                        x_list,
-                        {}
-                    ],
-                    'desc': [
-                        f'list birds or pigs',
-                    ]
-                },
-            }
-            self.game_train |= xxx
-
-        print('before add_config')
-        print(len(self.game_train))
-        add_config(self.game_train)
-        print('after add_config')
-        print(len(self.game_train))
-
-        print('final')
-        print(len(self.game_train))

games/angrybirds/config_const.py

@@ -1,27 +0,0 @@
-# constant
-
-# object verb
-x_change_position = 'change position'
-x_change_skin = 'change skin'
-
-# object
-x_bird = 'bird'
-x_pig = 'pig'
-x_house = 'house'
-x_obstacle = 'obstacle'
-x_pig_size = 'pig size'
-x_bird_size = 'bird size'
-
-# verb
-x_add = 'add'
-x_delete = 'delete'
-x_list = 'list'
-x_increase = 'increase'
-x_decrease = 'decrease'
-
-# attributes
-x_quantity = 'quantity'
-x_color = 'color'
-x_type = 'type'
-x_size = 'size'
-x_args = 'args'

games/angrybirds/test_all_angrybirds.py

@@ -1,305 +0,0 @@
-import pytest
-
-from games.angrybirds.angrybirds import angrybird
-
-status_OK = 'OK'
-status_Failure = 'Failure'
-bird = angrybird()
-bird.init_emb()
-
-def ai_test(inp,outp):
-    result = bird.query_emb(inp)
-    # result = bird.query_entry_point(inp)
-    assert result[0] == status_OK
-    # assert result[1][:2] == outp[:2], f"{inp}, {result}"
-    assert result[1] == outp, f"{inp}, {result}"
-
-def test_a00_test():
-    assert 1 + 1 == 2
-
-def test_a01_add_bird():
-    input = 'add a bird'
-    output = ['bird', 'add', {'quantity': 1}]
-    ai_test(input, output)
-
-def test_a02_i_want_add_birds():
-    input = 'I want add birds'
-    output = ['bird', 'add', {'quantity': 'some'}]
-    ai_test(input, output)
-
-def test_a03_reduce_bird_quantity():
-    input = 'Reduce a bird'
-    output = ['bird', 'delete', {'quantity': 1}]
-    ai_test(input, output)
-
-def test_a04_bird():
-    input = 'bird list'
-    output = ['bird', 'list', {}]
-    ai_test(input, output)
-
-def test_a05_show_birds():
-    input = 'show birds'
-    output = ['bird', 'list', {}]
-    ai_test(input, output)
-
-def test_a06_big_bird():
-    input = 'Big birds'
-    output = ['bird', 'add', {'type': 'big', 'quantity':'some'}]
-    ai_test(input, output)
-
-def test_a07_add_magnet_bird():
-    input = 'Add a Magnet Bird'
-    output = ['bird', 'add', {'type': 'magnet', 'quantity':1}]
-    ai_test(input, output)
-
-def test_a08_superheavy():
-    input = 'remove all superheavy birds'
-    output = ['bird', 'delete', {'type': 'superheavy', 'quantity':'all'}]
-    ai_test(input, output)
-
-def test_a09_catapult_bird():
-    input = 'add a Catapult bird'
-    output = ['bird', 'add', {'type': 'catapult','quantity':1}]
-    ai_test(input, output)
-
-def test_a10_flash_bird():
-    input = 'add 5 Flash Birds'
-    output = ['bird', 'add', {'type': 'flash', 'quantity':5}]
-    ai_test(input, output)
-
-def test_a11_antigravity_bird():
-    input = 'Anti-gravity bird'
-    output = ['bird', 'add', {'type': 'antigravity', 'quantity':1}]
-    ai_test(input, output)
-
-def test_a12_change_bird_picture():
-    input = "I want to change bird's picture"
-    output = ['change skin', {'args': 'birds'}]
-    ai_test(input, output)
-
-def test_a13_bird_photo():
-    input = 'bird photo'
-    output = ['change skin', {'args': 'birds'}]
-    ai_test(input, output)
-
-def test_a14_change_birds():
-    input = 'I want to change birds'
-    # output = ['change position', {'args': 'birds'}]
-    # strange semantics
-    output = ['bird', 'list', {}]
-    ai_test(input, output)
-
-def test_a15_add_a_pig():
-    input = 'add a pig'
-    output = ['pig', 'add', {'quantity': 1}]
-    ai_test(input, output)
-
-def test_a15_add_a_enemy():
-    input = 'add a enemy'
-    output = ['pig', 'add', {'quantity': 1}]
-    ai_test(input, output)
-
-def test_a16_add_a_pig():
-    input = 'Add a pig'
-    output = ['pig', 'add', {'quantity': 1}]
-    ai_test(input, output)
-
-def test_a16_add_a_enemy():
-    input = 'Add a enemy'
-    output = ['pig', 'add', {'quantity': 1}]
-    ai_test(input, output)
-
-def test_a17_reduce_a_pig():
-    input = 'reduce a pig'
-    output = ['pig', 'delete', {'quantity': 1}]
-    ai_test(input, output)
-
-def test_a17_reduce_a_enemy():
-    input = 'reduce a enemy'
-    output = ['pig', 'delete', {'quantity': 1}]
-    ai_test(input, output)
-
-def test_a18_list_pigs():
-    input = 'list pigs'
-    output = ['pig', 'list', {}]
-    ai_test(input, output)
-
-def test_a18_list_enemies():
-    input = 'list enemies'
-    output = ['pig', 'list', {}]
-    ai_test(input, output)
-
-def test_a19_elderly_pig():
-    input = 'add an Elderly pig'
-    output = ['pig', 'add', {'type': 'elderly', 'quantity': 1}]
-    ai_test(input, output)
-
-def test_a19_elderly_enemy():
-    input = 'add an Elderly enemy'
-    output = ['pig', 'add', {'type': 'elderly', 'quantity': 1}]
-    ai_test(input, output)
-
-def test_a20_sport_pig():
-    input = 'add a sport pig'
-    output = ['pig', 'add', {'type': 'sport', 'quantity': 1}]
-    ai_test(input, output)
-
-def test_a20_sport_enemy():
-    input = 'add a sport enemy'
-    output = ['pig', 'add', {'type': 'sport', 'quantity': 1}]
-    ai_test(input, output)
-
-def test_a21_pig_picture():
-    input = 'pig picture'
-    output = ['change skin', {'args': 'pigs'}]
-    ai_test(input, output)
-
-def test_a21_enemy_picture():
-    input = 'enemy picture'
-    output = ['change skin', {'args': 'pigs'}]
-    ai_test(input, output)
-
-def test_a22_make_the_pig_weaker():
-    input = 'make the pig weaker'
-    output = ['pig', 'decrease health points', {'quantity': 1}]
-    ai_test(input, output)
-
-def test_a22_make_the_enemy_weaker():
-    input = 'make the enemy weaker'
-    output = ['pig', 'decrease health points', {'quantity': 1}]
-    ai_test(input, output)
-
-def test_a23_make_the_pig_bigger():
-    input = 'make the pig bigger'
-    output = ['pig size', 'increase', {}]
-    ai_test(input, output)
-
-def test_a23_make_the_enemy_bigger():
-    input = 'make the enemy bigger'
-    output = ['pig size', 'increase', {}]
-    ai_test(input, output)
-
-def test_a24_change_pig_position():
-    input = "change pig's position"
-    output = ['pig', 'change position', {}]
-    ai_test(input, output)
-
-def test_a24_change_enemy_position():
-    input = "change enemy's position"
-    output = ['pig', 'change position', {}]
-    ai_test(input, output)
-
-def test_a25_move_the_pig():
-    input = 'move the pig'
-    output = ['pig', 'change position', {}]
-    ai_test(input, output)
-
-def test_a25_move_the_enemy():
-    input = 'move the enemy'
-    output = ['pig', 'change position', {}]
-    ai_test(input, output)
-
-def test_a26_change_obstacles():
-    input = 'I want to change the obstacles'
-    output = ['obstacle', 'change position', {}]
-    ai_test(input, output)
-
-def test_a27_():
-    input = 'background picture'
-    output = ['change background', {'args': 'picture'}]
-    ai_test(input, output)
-
-def test_a28_change_music():
-    input = 'change music'
-    output = ['change background', {'args': 'music'}]
-    ai_test(input, output)
-
-def test_a29_change_volumn():
-    input = 'change volumn'
-    output = ['change background', {'args': 'volumn'}]
-    ai_test(input, output)
-
-def test_a30_delete_3_pigs():
-    input = 'reduce 3 pigs'
-    output = ['pig', 'delete', {'quantity': 3}]
-    ai_test(input, output)
-
-def test_a30_delete_3_enemies():
-    input = 'reduce 3 enemies'
-    output = ['pig', 'delete', {'quantity': 3}]
-    ai_test(input, output)
-
-def test_a31_delete_5_birds():
-    input = 'remove five birds'
-    output = ['bird', 'delete', {'quantity': 5}]
-    ai_test(input, output)
-
-def test_a32_bird():
-    input = 'bird'
-    output = ['bird', 'list', {}]
-    ai_test(input, output)
-
-def test_a33_birds():
-    input = ' birds'
-    output = ['bird', 'list', {}]
-    ai_test(input, output)
-
-def test_a34_enemy():
-    input = 'enemy'
-    output = ['pig', 'list', {}]
-    ai_test(input, output)
-
-def test_a34_pig():
-    input = ' pig     '
-    output = ['pig', 'list', {}]
-    ai_test(input, output)
-
-def test_a34_enemy():
-    input = ' enemy     '
-    output = ['pig', 'list', {}]
-    ai_test(input, output)
-
-def test_a35_pigs():
-    input = ' piGs  \t'
-    output = ['pig', 'list', {}]
-    ai_test(input, output)
-
-def test_a35_enemies():
-    input = ' enemIEs  \t'
-    output = ['pig', 'list', {}]
-    ai_test(input, output)
-
-def test_a35_more_birds():
-    input = 'The game is too difficult. add more birds'
-    output = ['bird', 'add', {'quantity':'some'}]
-    ai_test(input, output)
-
-def test_a36_move_pigs():
-    input = 'move pigs'
-    output = ['pig', 'change position', {}]
-    ai_test(input, output)
-
-def test_a36_move_enemies():
-    input = 'move enemies'
-    output = ['pig', 'change position', {}]
-    ai_test(input, output)
-
-def test_a37_move_the_pigs():
-    input = 'move the pigs'
-    output = ['pig', 'change position', {}]
-    ai_test(input, output)
-
-def test_a37_move_the_enemies():
-    input = 'move the enemies'
-    output = ['pig', 'change position', {}]
-    ai_test(input, output)
-
-def test_a38_move_this_pig():
-    input = 'move this pig'
-    output = ['pig', 'change position', {}]
-    ai_test(input, output)
-
-def test_a38_move_this_enemy():
-    input = 'move this enemy'
-    output = ['pig', 'change position', {}]
-    ai_test(input, output)

games/angrybirds/test_delete_birds.py

@@ -1,103 +0,0 @@
-import pytest
-
-from games.angrybirds.angrybirds import angrybird
-
-status_OK = 'OK'
-status_Failure = 'Failure'
-bird = angrybird()
-bird.init_emb()
-
-def test_00_test():
-    assert 1 + 1 == 2
-
-def ai_test(inp,outp):
-    result = bird.query_emb(inp)
-    assert result[0] == status_OK
-    # assert result[1][:2] == outp[:2], f"{inp}, {result}"
-    assert result[1] == outp, f"{inp}, {result}"
-
-def test_delete_one_bird():
-    for bird_name in [
-        'big', 
-        'flash',
-    ]:
-        inp = f'remove a {bird_name} bird'
-        outp = ['bird', 'remove', {'type': bird_name}]
-        ai_test(inp, outp)        
-
-def test_delete_some_birds():
-    for bird_name in [
-        'big', 
-        'flash',
-    ]:
-        for bird_number, bird_number_numeric in {
-            'two': 2,
-            '2': 2,
-            'three': 3,
-            '3': 3,
-        }:
-            inp = f'remove {bird_number} {bird_name} bird'
-            outp = ['bird', 'remove', {'quantity': bird_number_numeric, 'type': bird_name}]
-            ai_test(inp, outp)        
-
-def test_delete_ranked_bird():
-    for bird_rank in [
-        'first', 
-        'second',
-        'third',
-    ]:
-        inp = f'remove the {bird_rank} bird'
-        outp = ['bird', 'remove', {'rank': bird_rank, 'quantity': 1}]
-        ai_test(inp, outp)        
-
-def test_delete_ranked_type_bird():
-    for bird_rank in [
-        'first', 
-        'second',
-        'third',
-    ]:
-        for bird_name in [
-            'big', 
-            'flash',
-        ]:
-            inp = f'remove the {bird_rank} {bird_name} bird'
-            outp = ['bird', 'remove', {'type': bird_name, 'rank': bird_rank, 'quantity': 1}]
-            ai_test(inp, outp)        
-
-def test_delete_ranked_birds():
-    for bird_rank in [
-        'first', 
-        'last',
-    ]:
-        for bird_number, bird_number_numeric in {
-            'two': 2,
-            '2': 2,
-            'three': 3,
-            '3': 3,
-        }:
-            inp = f'remove the {bird_rank} {bird_number} birds'
-            outp = ['bird', 'remove', {'rank': bird_rank, 'quantity': bird_number_numeric}]
-            ai_test(inp, outp)        
-
-def test_delete_ranked_type_birds():
-    for bird_rank in [
-        'first', 
-        'last',
-    ]:
-        for bird_number, bird_number_numeric in {
-            'two': 2,
-            '2': 2,
-            'three': 3,
-            '3': 3,
-        }:
-            for bird_name in [
-                'big', 
-                'flash',
-            ]:
-                inp = f'remove the {bird_rank} {bird_number} {bird_name} birds'
-                outp = ['bird', 'remove', {
-                    'rank': bird_rank, 
-                    'quantity': bird_number_numeric,
-                    'type': bird_name,
-                }]
-                ai_test(inp, outp)        

games/angrybirds/test_parameter.py

@@ -1,26 +0,0 @@
-import pytest
-
-from games.angrybirds.angrybirds import angrybird
-
-status_OK = 'OK'
-status_Failure = 'Failure'
-bird = angrybird()
-bird.init_emb()
-
-def test_00_test():
-    assert 1 + 1 == 2
-
-def ai_test(inp,outp):
-    result = bird.query_emb(inp)
-    assert result[0] == status_OK
-    # assert result[1][:2] == outp[:2], f"{inp}, {result}"
-    assert result[1] == outp, f"{inp}, {result}"
-
-def test_delete_one_bird():
-    for bird_name in [
-        'big', 
-        'flash',
-    ]:
-        inp = f'remove a {bird_name} bird'
-        outp = ['bird', 'remove', {'type': bird_name}]
-        ai_test(inp, outp)        

games/game.py

@@ -1,129 +0,0 @@
-import json
-from datetime import datetime
-from zoneinfo import ZoneInfo
-import glob
-
-from openai.embeddings_utils import get_embedding as openai_get_embedding
-from openai.embeddings_utils import cosine_similarity as openai_cosine_similarity
-
-debug = 2
-
-class game:
-    status_failure = 'Failure'
-    status_OK = 'OK'
-
-    ai_version   = ''
-    data_version = ''
-
-    def __init__(self):
-        pass
-
-    def query_entry_point(self, command: str):
-        status, res, desc, results = \
-            self.query_emb(command)
-
-        t = datetime.now()
-        t = t.astimezone(ZoneInfo('Asia/Shanghai'))
-        # t = t.replace(microsecond=0)
-
-        res1 = {
-            'status': status,
-            'result': res,
-            'desc': desc,
-            'command': command,
-            # 'score': f'{top_sim:.4f}',
-            'timestamp': f"{t}",
-            'ai-version': self.ai_version,
-        }
-
-        res2 = []
-        for i, e in enumerate(results[:]):
-            tmp = [i, f"{e[1]:.8f}", e[0]]
-            res2.append(tmp)
-
-        self.query_count += 1
-        print()
-        print('-'*20)
-        print(f'[{self.PATH}] query count: {self.query_count}')
-        print(f'input: {command}')
-        print(json.dumps(res1, indent=2))
-        print(json.dumps(res2, indent=2))
-        print('%'*20)
-        print(f'[{self.PATH}] query count: {self.query_count}')
-        print(f'input: {command}')
-        print(res1['status'])
-        print(f'top: {results[0][0]}')
-        print('-'*20)
-        print()
-        return res1, res2
-
-
-    def list_functionality(self):
-        s = []
-        for cmd in self.train:
-            desc = self.train[cmd]['desc'][0]
-            if desc not in s:
-                s.append(desc)
-        res = {
-            'version': self.data_version,
-            'data': [f'({i+1}) {e}' for i, e in enumerate(sorted(s))],
-        }
-        return res
-
-
-    def show_json_old(self):
-        s = {}
-        for cmd in self.train:
-            s[cmd] = {}
-            s[cmd]['result'] = self.train[cmd]['result']
-        res = {
-            'version': self.data_version,
-            'data': s,
-        }
-        return res
-
-
-    def show_json(self):
-        s = {}
-        for cmd in self.train:
-            desc = self.train[cmd]['desc'][0]
-            result = self.train[cmd]['result']
-            if desc not in s:
-                s[desc] = [result,]
-            else:
-                if result not in s[desc]:
-                    s[desc].append(result)        
-        res = {
-            'version': self.data_version,
-            'data': s,
-        }
-        return res
-
-
-    def show_json_x(self):
-        s = {}
-        for cmd in self.train:
-            key = json.dumps(self.train[cmd]['result'])
-            desc = set(self.train[cmd]['desc'])
-            if key not in s.keys():
-                s[key] = desc
-            else:
-                s[key] |= desc
-        res = {
-            'version': self.data_version,
-            'data': s,
-        }
-
-        res2 = {}
-        i = 0
-        for key in res['data'].keys():
-            res2 |= {i: [json.loads(key), list(res['data'][key])]}
-            i += 1
-
-        res = {
-            'version': self.data_version,
-            'data': res2,
-        }
-
-        return res
-

games/game_angrybird.py

@@ -1,23 +0,0 @@
-import os
-import json
-import requests
-
-from games.angrybirds.angrybirds import angrybird
-
-
-game = angrybird()
-
-
-def process(command: str):
-    url = f'https://{os.getenv("hfs_space")}.hf.space/games/angrybird'
-    # url = 'http://127.0.0.1:8080/nlp'
-    myobj = {
-        'command': command,
-    }
-    headers = {'Authorization': f'Bearer {os.getenv("hfs_token")}'}
-
-    x = requests.post(url, json=myobj, headers=headers)
-    print(x.text)
-
-    # res1, _ = game.query_entry_point(command)
-    return json.loads(x.text)

games/game_snake.py

@@ -1,23 +0,0 @@
-import os
-import json
-import requests
-
-from games.snake.snake import snake
-
-
-game = snake()
-
-
-def process(command: str):
-    url = f'https://{os.getenv("hfs_space")}.hf.space/games/snake'
-    # url = 'http://127.0.0.1:8080/nlp'
-    myobj = {
-        'command': command,
-    }
-    headers = {'Authorization': f'Bearer {os.getenv("hfs_token")}'}
-
-    x = requests.post(url, json=myobj, headers=headers)
-    print(x.text)
-
-    # res1, _ = game.query_entry_point(command)
-    return json.loads(x.text)

games/game_util.py

@@ -1,16 +0,0 @@
-from fastapi import HTTPException, status
-
-import games.game_snake as game_snake
-import games.game_angrybird as game_angrybird
-
-
-GAME_ANGRY_BIRD = 'angrybird'
-GAME_SNAKE = 'snake'
-
-
-def nlp(game: str, token: str, command: str):
-    if game == GAME_ANGRY_BIRD:
-        return game_angrybird.process(token, command)
-    if game == GAME_SNAKE:
-        return game_snake.process(token, command)
-    raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)

games/snake/snake.py

@@ -1,37 +0,0 @@
-import json
-from pathlib import Path
-
-import openai
-from games.game import game
-# from games.game_openai import api_key
-
-debug = 2
-
-class snake (game):
-    ai_version   = '1.4.0.2023-12-29-a' # fastapi
-    data_version = ai_version
-
-    def __init__(self):
-        # openai.api_key = api_key
-        self.ROOT = Path(__file__).parent.parent.parent
-        self.PATH = self.__class__.__name__
-        self.embeddings = {}
-        self.query_count = 0
-
-        def init_train():
-            PATH = Path(__file__).parent
-            with open(PATH/"game_train.json", 'r') as f:
-                unnormalized = json.load(f)
-
-            self.train = {}
-            for cmd in unnormalized.keys():
-                key = cmd.strip().lower()
-                self.train[key] = unnormalized[cmd]
-            print(f"{len(self.train)=}")
-
-        # self.train = self.init_train(config)
-        init_train()
-        
-
-    def special_case(self, command: str):
-        return command

main.py

@@ -1,14 +0,0 @@
-from fastapi import FastAPI, Body
-
-import games.game_util as game_util
-
-app = FastAPI()
-
-
-@app.get("/")
-async def root():
-    return {"message": "PGAI API"}
-
-@app.post("/nlp")
-def test(game: str=Body(), token: str=Body(), command: str=Body()):
-    return game_util.nlp(game, token, command)

requirements.txt

@@ -1,10 +1,13 @@
 fastapi
-uvicorn[standard]
-openai==0.27.8
-matplotlib==3.7.2
-matplotlib-inline==0.1.6
-plotly==5.15.0
-pandas==2.0.3
-pandas-stubs==2.1.1.230928
-scipy==1.11.4
-scikit-learn==1.3.0
+huggingface-hub
+langchain
+langchain-community
+langchain-huggingface
+langchain_openai
+transformers
+sentence-transformers
+faiss-cpu
+dashscope
+pdfminer.six
+pdfplumber
+# git+https://github.com/east-and-west-magic/pgsoft.git@tag-2024-01-11-a

src/fileservice.py

@@ -0,0 +1,139 @@
+from fastapi import APIRouter
+from pgsoft.pgdate.date_utils import beijing
+from pgsoft.pghash.md5 import md5
+import pgsoft.pgfile as pgfile
+from time import sleep
+import json
+import os
+
+router = APIRouter(prefix="/file", tags=["File Service"])
+dataset_id = "pgsoft/game"
+tempdir = "game"
+pgai_code = os.environ.get("pgai_code")
+db_token = os.environ.get("db_token")
+if db_token:
+    print(db_token[:5])
+    
+
+@router.get("/download")
+def download_file(game: str, token: str, gamecode: str):
+    if token != pgai_code:
+        print(f"[{beijing()}][download file] failed")
+        return {"status": "Failure", "detail": "Invalid token"}
+    game = game.strip().lower()
+    filename = gamecode.strip() + ".json"
+    remotepath = "/".join([game, filename[:2], filename])
+    res = pgfile.download(
+        dataset_id,
+        remotepath=remotepath,
+        repo_type="dataset",
+        localdir=tempdir,
+        token=db_token,
+    )
+    if not res:
+        print(f"[{beijing()}][download file] failed")
+        return {"status": "Failure", "detail": "File not found or server error"}
+    with open(res, "r") as f:
+        outp = json.load(f)
+    print(f"[{beijing()}][download file] OK")
+    return {"status": "OK", "result": outp}
+
+
+@router.post("/upload")
+def upload_file(game: str, token: str, content: str):
+    if token != pgai_code:
+        print(f"[{beijing()}][upload file] failed")
+        return {"status": "Failure", "detail": "Invalid token"}
+    game = game.strip().lower()
+    try:
+        content_dict = json.loads(content)
+    except json.JSONDecodeError as e:
+        print(f"[{beijing()}][upload file] failed, {type(e)}: {e}")
+        return {"status": "Failure", "detail": "Invalid JSON"}
+
+    if not isinstance(content_dict, dict):
+        print(f"[{beijing()}][upload file] failed, not a dict")
+        return {"status": "Failure", "detail": "not a dict"}
+
+    needed_keys = ["game-file", "device-id"]
+    for key in needed_keys:
+        if key not in content_dict:
+            print(f'[{beijing()}][upload file] failed, missed "{key}"')
+            return {"status": "Failure", "detail": f'missed "{key}"'}
+
+    if not isinstance(content_dict["device-id"], str):
+        print(f'[{beijing()}][upload file] failed, "device-id" is not a str')
+        return {"status": "Failure", "detail": '"device-id" is not a str'}
+    if not isinstance(content_dict["game-file"], dict):
+        print(f'[{beijing()}][upload file] failed, "game-file" is not a dict')
+        return {"status": "Failure", "detail": '"game-file" is not a dict'}
+
+    obj = {
+        "upload-time": beijing().__str__(),
+        "game-file": content_dict["game-file"],
+    }
+    maxtry = 5
+    for retry in range(maxtry):
+        md5code = md5(obj)
+        remotepath = "/".join([game, md5code[:2], md5code + ".json"])
+        if not pgfile.api.file_exists(
+            repo_id=dataset_id,
+            filename=remotepath,
+            repo_type="dataset",
+            token=db_token,
+        ):
+            break
+        sleep(0.1)
+        obj["upload-time"] = beijing().__str__()
+        maxtry -= 1
+    if not maxtry and pgfile.api.file_exists(
+        repo_id=dataset_id,
+        filename=remotepath,
+        repo_type="dataset",
+        token=db_token,
+    ):
+        print(f"[{beijing()}][upload file] failed, timeout, please retry")
+        return {"status": "Failure", "detail": "timeout, please retry"}
+    filedir = os.path.join(tempdir, game, md5code[:2])
+    if not os.path.exists(filedir):
+        os.makedirs(filedir)
+    filepath = os.path.join(filedir, md5code + ".json")
+    content_indented = json.dumps(content_dict, indent=4)
+    with open(filepath, "w") as f:
+        f.write(content_indented)
+    res = pgfile.upload(
+        filepath,
+        remotepath,
+        dataset_id,
+        "dataset",
+        db_token,
+        f"Updated at {beijing()}",
+    )
+    if not res:
+        print(f"[{beijing()}][upload file] failed")
+        return {"status": "Failure", "detail": "server error"}
+    print(f"[{beijing()}][upload file] OK")
+    return {"status": "OK", "result": md5code}
+
+
+@router.get("/list")
+def list_files(game: str, token: str):
+    if token != pgai_code:
+        print(f"[{beijing()}][list files] failed")
+        return {"status": "Failure", "detail": "Invalid token"}
+    game = game.strip().lower()
+    games = pgfile.list_files(
+        repo_id=dataset_id,
+        repo_type="dataset",
+        token=db_token,
+    )
+    if games is None:
+        print(f"[{beijing()}][list files] failed")
+        return {"status": "Failure", "detail": "server error"}
+    games = {
+        item.split(".")[0][-32:]: item
+        for item in games
+        if item.endswith(".json") and item.startswith(game)
+    }
+    print(f"[{beijing()}][list files] OK")
+    return {"status": "OK", "result": games}

src/main.py

@@ -0,0 +1,12 @@
+from fastapi import FastAPI
+from . import fileservice
+from . import talk_to_your_manual
+
+app = FastAPI()
+# app.include_router(fileservice.router)
+app.include_router(talk_to_your_manual.router)
+
+
+@app.get("/")
+def root():
+    return {"message": "Server for Prove of Concept"}

src/talk_to_your_manual/__init__.py

@@ -0,0 +1,11 @@
+from fastapi import APIRouter
+from . import use_aliyun
+from . import use_huggingface
+
+
+router = APIRouter(
+    prefix="/talk-to-your-manual",
+    tags=["Talk To Your Manual"],
+)
+router.include_router(use_aliyun.router)
+# router.include_router(use_huggingface.router)

src/talk_to_your_manual/use_aliyun.py

@@ -0,0 +1,26 @@
+from dashscope import Application
+from fastapi import APIRouter
+from fastapi.responses import JSONResponse
+import os
+
+router = APIRouter()
+API_KEY = os.environ.get("API_KEY")
+APP_ID = os.environ.get("APP_ID")
+
+
+@router.get("/aliyun")
+async def call_aliyun(prompt: str):
+    response = Application.call(
+        api_key=API_KEY,
+        app_id=APP_ID,
+        prompt=prompt,
+    )
+    return JSONResponse(
+        status_code=response.status_code,
+        content={
+            "request_id": response.request_id,
+            "message": response.message,
+            "output": response.output,
+            "usage": response.usage,
+        },
+    )

src/talk_to_your_manual/use_huggingface.py

@@ -0,0 +1,85 @@
+from transformers import BloomForCausalLM, BloomTokenizerFast
+from langchain.chains import RetrievalQA
+from langchain_community.vectorstores import FAISS
+from langchain.text_splitter import CharacterTextSplitter
+from langchain_community.document_loaders import PDFPlumberLoader
+from langchain_huggingface import HuggingFaceEndpointEmbeddings
+from langchain_huggingface import HuggingFacePipeline
+from transformers import pipeline
+from fastapi import APIRouter, status
+from fastapi.responses import JSONResponse
+import os
+
+router = APIRouter()
+
+# 查询文本
+query = "游戏的主要玩法是什么？"
+max_new_tokens = 100
+
+# 加载模型和tokenizer
+db_token = os.environ.get("db_token")
+model = BloomForCausalLM.from_pretrained(
+    "bigscience/bloom-560m",
+    token=db_token,
+)
+tokenizer = BloomTokenizerFast.from_pretrained(
+    "bigscience/bloom-560m",
+    token=db_token,
+)
+
+# 加载文档内容
+loader = PDFPlumberLoader("建造大楼游戏说明.pdf")
+documents = loader.load()
+splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
+documents = splitter.split_documents(documents)
+# 使用 Sentence-BERT 模型将文档转化为向量
+embeddings = HuggingFaceEndpointEmbeddings(
+    repo_id="sentence-transformers/all-MiniLM-L6-v2",
+    huggingfacehub_api_token=db_token,
+)
+# 使用 FAISS 创建索引
+faiss_index = FAISS.from_documents(documents, embeddings)
+# 保存索引
+faiss_index.save_local("faiss_index")
+# 从 FAISS 索引中检索相关文档
+retriever = faiss_index.as_retriever()
+
+# 创建 Hugging Face Pipeline 以使用 BLOOM 模型
+hf_pipeline = pipeline(
+    "text-generation",
+    model=model,
+    tokenizer=tokenizer,
+    max_new_tokens=max_new_tokens,
+)
+
+# 使用 LangChain 的 HuggingFacePipeline 类
+llm = HuggingFacePipeline(pipeline=hf_pipeline)
+
+# 创建问答链
+qa_chain = RetrievalQA.from_chain_type(
+    llm=llm,
+    retriever=retriever,
+    return_source_documents=True,
+)
+
+
+@router.get("/huggingface")
+async def call_huggingface(prompt: str):
+    try:
+        response = qa_chain.invoke({"query": prompt})
+        text: str = response["result"]
+        text = text.split("\nHelpful Answer: ")[1]
+        text = text.split("\n\n")[0]
+        return JSONResponse(
+            status_code=status.HTTP_200_OK,
+            content={
+                "result": text,
+            },
+        )
+    except Exception as e:
+        return JSONResponse(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            content={
+                "error": str(e),
+            },
+        )