connect to hf

games/angrybirds/angrybirds.py

@@ -4,7 +4,7 @@ from collections import Counter
 
 import openai
 from games.game import game
-from games.game_openai import api_key
+# from games.game_openai import api_key
 from games.angrybirds.config import config
 
 debug = 2
@@ -14,7 +14,7 @@ class angrybird (game):
     data_version = ai_version
 
     def __init__(self):
-        openai.api_key = api_key
+        # openai.api_key = api_key
         self.ROOT = Path(__file__).parent.parent.parent
         self.PATH = self.__class__.__name__
         self.embeddings = {}

games/game_openai.py

@@ -1,2 +0,0 @@
-
-api_key = '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'

games/game_snake.py

@@ -6,13 +6,16 @@ from games.snake.snake import snake
 
 game = snake()
 
-def process(command: str):
-    url = 'http://127.0.0.1:8080/nlp'
+def process(token: str, command: str):
+    url = 'https://steveagi-pgtrain.hf.space/games/snake'
+    # url = 'http://127.0.0.1:8080/nlp'
     myobj = {
-        'game': 'snake',
         'command': command,
     }
-    x = requests.post(url, json=myobj)
+    auth_token = token
+    headers = {'Authorization': f'Bearer {auth_token}'}
+
+    x = requests.post(url, json=myobj, headers=headers)
     print(x.text)
 
     # res1, _ = game.query_entry_point(command)

games/game_util.py

@@ -8,9 +8,9 @@ GAME_ANGRY_BIRD = 'angrybird'
 GAME_SNAKE = 'snake'
 
 
-def nlp(game: str, command: str):
+def nlp(game: str, token: str, command: str):
     if game == GAME_ANGRY_BIRD:
-        return game_angrybird.process(command)
+        return game_angrybird.process(token, command)
     if game == GAME_SNAKE:
-        return game_snake.process(command)
+        return game_snake.process(token, command)
     raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)

games/snake/snake.py

@@ -3,7 +3,7 @@ from pathlib import Path
 
 import openai
 from games.game import game
-from games.game_openai import api_key
+# from games.game_openai import api_key
 
 debug = 2
 
@@ -12,7 +12,7 @@ class snake (game):
     data_version = ai_version
 
     def __init__(self):
-        openai.api_key = api_key
+        # openai.api_key = api_key
         self.ROOT = Path(__file__).parent.parent.parent
         self.PATH = self.__class__.__name__
         self.embeddings = {}

main.py

@@ -10,5 +10,5 @@ async def root():
     return {"message": "PGAI API"}
 
 @app.post("/nlp")
-def test(game: str=Body(), command: str=Body()):
-    return game_util.nlp(game, command)
+def test(game: str=Body(), token: str=Body(), command: str=Body()):
+    return game_util.nlp(game, token, command)