Bladeren bron

fix: dataset destination error (#477)

tags/0.3.7
John Wang 2 jaren geleden
bovenliggende
commit
9d98669e7d
No account linked to committer's email address
1 gewijzigde bestanden met toevoegingen van 12 en 4 verwijderingen
  1. 12
    4
      api/core/chain/multi_dataset_router_chain.py

+ 12
- 4
api/core/chain/multi_dataset_router_chain.py Bestand weergeven

import math import math
import re
from typing import Mapping, List, Dict, Any, Optional from typing import Mapping, List, Dict, Any, Optional


from langchain import PromptTemplate from langchain import PromptTemplate


route = self.router_chain.route(inputs) route = self.router_chain.route(inputs)


if not route.destination:
destination = ''
if route.destination:
pattern = r'\b[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}\b'
match = re.search(pattern, route.destination, re.IGNORECASE)
if match:
destination = match.group()

if not destination:
return {"text": ''} return {"text": ''}
elif route.destination in self.dataset_tools:
return {"text": self.dataset_tools[route.destination].run(
elif destination in self.dataset_tools:
return {"text": self.dataset_tools[destination].run(
route.next_inputs['input'] route.next_inputs['input']
)} )}
else: else:
raise ValueError( raise ValueError(
f"Received invalid destination chain name '{route.destination}'"
f"Received invalid destination chain name '{destination}'"
) )

Laden…
Annuleren
Opslaan