Merge pull request #19 from Gentopia-AI/feature/packaging-for-gentpool

Packaging Gentopia

Author: liuxukun2000

gentopia/assembler/loader.py

@@ -43,4 +43,4 @@ def file(self, node: yaml.Node) -> Any:
         if not filename.is_absolute():
             filename = self._root / filename
         with open(filename, 'r') as f:
-            return f.read()
+            return f.read().strip()

gentopia/llm/client/openai.py

@@ -117,7 +117,6 @@ def function_chat_completion(self, message: List[dict],
                 presence_penalty=self.params.presence_penalty,
             )
             response_message = response.choices[0]["message"]
-            print("!!!!!", response_message)
 
             if response_message.get("function_call"):
                 function_name = response_message["function_call"]["name"]

main.yaml

@@ -44,4 +44,4 @@ plugins:
 
 # Authentication
 auth:
-  OPENAI_API_KEY: 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
+  OPENAI_API_KEY: !file /home/billxbf/Documents/myks/openai.key

requirements.txt

@@ -0,0 +1,14 @@
+fastapi==0.98.0
+geopy==2.3.0
+langchain==0.0.209
+optimum==1.8.6
+peft==0.4.0.dev0
+pydantic==1.10.9
+pytest==7.3.2
+PyYAML==6.0
+PyYAML==6.0
+Requests==2.31.0
+setuptools==67.8.0
+torch==2.0.1
+transformers==4.31.0.dev0
+uvicorn==0.22.0

setup.py

@@ -0,0 +1,29 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='gentopia',
+    version='0.1',
+    packages=find_packages(),
+    url='https://github.com/Gentopia-AI/Gentopia',
+    license='Apache-2.0 license',
+    author='billxbf',
+    author_email='billxbf@gmail.com',
+    description='Gentopia assembles and serves specialized ALM agents driven by configs.',
+    long_description=open('README.md').read(),
+    long_description_content_type='text/markdown',
+    install_requires=[
+        'fastapi',
+        'geopy',
+        'langchain',
+        'optimum',
+        'peft',
+        'pydantic',
+        'pytest',
+        'PyYAML',
+        'requests',
+        'setuptools',
+        'torch',
+        'transformers',
+        'uvicorn',
+    ],
+)

test.py

@@ -33,13 +33,13 @@ def print_tree(obj, indent=0):
 if __name__ == '__main__':
     mp.set_start_method('spawn', force=True)
     config = Config.load('main.yaml')
-    print(config)
+    #print(config)
     # exit(0)
     assembler = AgentAssembler(file='main.yaml')
     #assembler.manager = LocalLLMManager()
     agent = assembler.get_agent()
     ans = ""
-    response = agent.run("what is the square root of 23?")
+    response = agent.run("who is Leonardo da Vinci's first wife?")
     print(response)
     # for i in agent.llm.stream_chat_completion("what is the square root of 23?"):
     #     print(i)