Omkar008 commited on
Commit
b009d8f
1 Parent(s): 3f4f1de

Update services/utils.py

Browse files
Files changed (1) hide show
  1. services/utils.py +4 -3
services/utils.py CHANGED
@@ -63,7 +63,7 @@ def strcuture_document_data(raw_text:str)->dict:
63
  input_variables=["query"],
64
  partial_variables={"format_instructions": parser.get_format_instructions()},
65
  )
66
- print(parser.get_format_instructions())
67
  input = prompt.format_prompt(query=doc_query)
68
  with get_openai_callback() as cb:
69
  result = model(input.to_string())
@@ -76,8 +76,9 @@ def strcuture_document_data(raw_text:str)->dict:
76
  class_object= parser.parse(result)
77
  dict_object=class_object.__dict__
78
  if not dict_object:
79
- null_values+=1
80
- print(null_values)
 
81
  print("printing structured json")
82
  print(dict_object)
83
  return dict_object
 
63
  input_variables=["query"],
64
  partial_variables={"format_instructions": parser.get_format_instructions()},
65
  )
66
+ # print(parser.get_format_instructions())
67
  input = prompt.format_prompt(query=doc_query)
68
  with get_openai_callback() as cb:
69
  result = model(input.to_string())
 
76
  class_object= parser.parse(result)
77
  dict_object=class_object.__dict__
78
  if not dict_object:
79
+ print(dict_object)
80
+ print("Got null for dict object")
81
+
82
  print("printing structured json")
83
  print(dict_object)
84
  return dict_object