gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Remove setting envvars from arg

Browse source
This commit is contained in:
Rita Aleksziev 2025-01-08 12:33:14 +01:00
parent 8ffef5034a
commit f4397bf940
1 changed files with 0 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
examples/python/code_graph_example.py
View file

@ -24,18 +24,6 @@ def parse_args():
default=True,
help="Whether or not to process non-code files"
)
parser.add_argument(
"--mock_embedding",
type=lambda x: x.lower() in ("true", "1"),
default=True,
help="Whether or not to mock embedding and code summary"
)
parser.add_argument(
"--mock_code_summary",
type=lambda x: x.lower() in ("true", "1"),
default=True,
help="Whether or not to mock code summary"
)
parser.add_argument(
"--time",
type=lambda x: x.lower() in ("true", "1"),
@ -47,18 +35,8 @@ def parse_args():
if __name__ == "__main__":
setup_logging(logging.ERROR)
import os
args = parse_args()
if args.mock_embedding:
os.environ["MOCK_EMBEDDING"] = "true"
print("Mocking embedding.")
if args.mock_code_summary:
os.environ["MOCK_CODE_SUMMARY"] = "true"
print("Mocking code summary.")
if args.time:
import time
start_time = time.time()