Pipalskill commited on
Commit
c53ba71
·
verified ·
1 Parent(s): 6ab13ea

Update agent_langchain.py

Browse files
Files changed (1) hide show
  1. agent_langchain.py +2 -0
agent_langchain.py CHANGED
@@ -7,6 +7,7 @@ os.environ["TORCH_HOME"] = "/tmp/torch"
7
 
8
  import requests
9
  import torch
 
10
  import time
11
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
12
  import numpy as np
@@ -28,6 +29,7 @@ GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
28
  ROUTING_URL = os.environ.get("ROUTING_URL")
29
  SPACE_URL = os.environ.get("SPACE_URL", "http://localhost:7860")
30
  FIREBASE_CREDS_PATH = os.environ.get("FIREBASE_CREDS_PATH")
 
31
 
32
  # Initialize Firebase
33
  db = None
 
7
 
8
  import requests
9
  import torch
10
+ import json
11
  import time
12
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
13
  import numpy as np
 
29
  ROUTING_URL = os.environ.get("ROUTING_URL")
30
  SPACE_URL = os.environ.get("SPACE_URL", "http://localhost:7860")
31
  FIREBASE_CREDS_PATH = os.environ.get("FIREBASE_CREDS_PATH")
32
+ firebase_creds_json = os.getenv("FIREBASE_CREDS_JSON")
33
 
34
  # Initialize Firebase
35
  db = None