joker7094 commited on
Commit
67ea255
·
1 Parent(s): 3e9ccc5

fixing issues related to sentence-transformers

Browse files
Files changed (1) hide show
  1. query_processing.py +2 -3
query_processing.py CHANGED
@@ -61,8 +61,7 @@ def get_llm():
61
  if not api_key:
62
  raise ValueError("GOOGLE_API_KEY not found in environment variables")
63
 
64
- # _llm = ChatGoogleGenerativeAI(model="gemma-3-27b-it", google_api_key=api_key)
65
- _llm = ChatGoogleGenerativeAI(model="gemma-3-27b-it", google_api_key=AIzaSyBjWFgLWn3zZLKmTf1_EcVD9FY86l4NDpM)
66
  prompt_template = ChatPromptTemplate.from_template(
67
  """
68
  You are a helpful assistant designed to analyze job descriptions and extract key information.
@@ -87,7 +86,7 @@ def get_llm():
87
  return f"Analysis unavailable (API error). Original text: {text[:500]}"
88
 
89
  _llm_chain = ({"context": RunnablePassthrough()}
90
- | (lambda x: {"result": fallback_processor(x["context"])})
91
  | (lambda x: x["result"]))
92
  return _llm_chain
93
 
 
61
  if not api_key:
62
  raise ValueError("GOOGLE_API_KEY not found in environment variables")
63
 
64
+ _llm = ChatGoogleGenerativeAI(model="gemma-3-27b-it", google_api_key=api_key)
 
65
  prompt_template = ChatPromptTemplate.from_template(
66
  """
67
  You are a helpful assistant designed to analyze job descriptions and extract key information.
 
86
  return f"Analysis unavailable (API error). Original text: {text[:500]}"
87
 
88
  _llm_chain = ({"context": RunnablePassthrough()}
89
+ | (lambda x: {"context": x["context"], "result": fallback_processor(x["context"])})
90
  | (lambda x: x["result"]))
91
  return _llm_chain
92