Fix fewshot examples (neo4j#54)

Author: tomasonjo

api/src/fewshot_examples.py

@@ -1,12 +1,5 @@
-import os
-
-openai_api_key = os.environ.get("OPENAI_API_KEY", "")
-
-fewshot_examples = {}
-
-fewshot_examples[
-    "companies"
-] = f"""
+def get_fewshot_examples(openai_api_key):
+    return f"""
 #How is Emil Eifrem connected to Michael Hunger?
 MATCH (p1:Person {{name:"Emil Eifrem"}}), (p2:Person {{name:"Michael Hunger"}})
 MATCH p=shortestPath((p1)-[*]-(p2))

api/src/main.py

@@ -15,7 +15,7 @@
 from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
-from fewshot_examples import fewshot_examples
+from fewshot_examples import get_fewshot_examples
 from llm.openai import OpenAIChat
 from pydantic import BaseModel
 
@@ -141,7 +141,7 @@ async def onToken(token):
             text2cypher = Text2Cypher(
                 database=neo4j_connection,
                 llm=default_llm,
-                cypher_examples=fewshot_examples["companies"],
+                cypher_examples=get_fewshot_examples(api_key),
             )
 
             if "type" not in data: