mtyrrell commited on
Commit
32ef85f
·
1 Parent(s): 4ef25e0

ts startup

Browse files
Files changed (1) hide show
  1. app/nodes.py +1 -2
app/nodes.py CHANGED
@@ -21,7 +21,6 @@ config = getconfig("params.cfg")
21
  RETRIEVER = config.get("retriever", "RETRIEVER")
22
  GENERATOR = config.get("generator", "GENERATOR")
23
  INGESTOR = config.get("ingestor", "INGESTOR")
24
- GEOJSON_INGESTOR = config.get("ingestor", "GEOJSON_INGESTOR")
25
  MAX_CONTEXT_CHARS = int(config.get("general", "MAX_CONTEXT_CHARS"))
26
 
27
  retriever_adapter = RetrieverAdapter("params.cfg")
@@ -65,7 +64,7 @@ def ingest_node(state: GraphState) -> GraphState:
65
 
66
  try:
67
  # Choose ingestor based on file type
68
- ingestor_url = GEOJSON_INGESTOR if file_type == "geojson" else INGESTOR
69
  logger.info(f"Using ingestor: {ingestor_url}")
70
 
71
  client = Client(ingestor_url, hf_token=os.getenv("HF_TOKEN"))
 
21
  RETRIEVER = config.get("retriever", "RETRIEVER")
22
  GENERATOR = config.get("generator", "GENERATOR")
23
  INGESTOR = config.get("ingestor", "INGESTOR")
 
24
  MAX_CONTEXT_CHARS = int(config.get("general", "MAX_CONTEXT_CHARS"))
25
 
26
  retriever_adapter = RetrieverAdapter("params.cfg")
 
64
 
65
  try:
66
  # Choose ingestor based on file type
67
+ ingestor_url = INGESTOR
68
  logger.info(f"Using ingestor: {ingestor_url}")
69
 
70
  client = Client(ingestor_url, hf_token=os.getenv("HF_TOKEN"))